OOM Error when I am trying to run FM on dataset ml-tag for the best performance

[hexiangnan]

OOM Error when I am trying to run FM on dataset ml-tag for the best performance

[Dellen]

What's strange is, isn't that that number should be the batch_size? I don't understand why the total number of training sample would show up.

[Dellen]

Found it. It was because in the init phase, it would run the whole dataset through sess.run(). But now I wonder why the code for AFM works fine, because it would do the same in the init phase.

[Dellen]

Now I understand. In AFM, it doesn't need to do the square calculation.
self.summed_features_emb = tf.reduce_sum(self.nonzero_embeddings, 1, keep_dims=True) # None * 1 * K

Now the problem is how can I run FM on my computer, 8GB GPU. What is the your environment setup for this?

[Dellen]

I solved this problem by changing the evaluate function into feeding the data in batches. Now it works fine.