OOM when allocating tensor with shape[2259000,300]
kepsail opened this issue · comments
kepsail commented
Caused by op 'gradients/concat', defined at:
File "main.py", line 34, in
Config.epochs, train, e, Config.k, test)
File ".\include\Model.py", line 275, in training
train_step = tf.train.AdamOptimizer(learning_rate).minimize(loss)