Weirdly enough it stopped training at 0.69 epoch (no, I am not making this up!). Will have to debug why.
The training was kind of unstable in the beginning but I lowered the lr and added max_grad_norm. However I feel the lr is too low and could have got better results with higher lr.
Training hyperparameters
The following hyperparameters were used during training:
learning_rate: 1e-05
train_batch_size: 2
eval_batch_size: 4
seed: 42
distributed_type: multi-GPU
num_devices: 2
total_train_batch_size: 4
total_eval_batch_size: 8
optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
