epoch not working, pytorch issue?

[ChloeJKim]

Hello I'm having this problem, can you help me?

after I run this command
python run_bert_triple_classifier.py --task_name kg --do_train --do_eval --do_predict --data_dir data/WN11 --bert_model bert-base-uncased --max_seq_length 20 --train_batch_size 32 --learning_rate 5e-5 --num_train_epochs 3.0 --output_dir output_WN11/ --gradient_accumulation_steps 1 --eval_batch_size 512

I get the following error where the epoch doesn't start

Thank you in advance

[ChloeJKim]

I also have torch installed