I did some modifications in TrainingArguments. I commented the max_steps and set training epochs as 2. But when we load this args in Trainer and train the model, it generates this weird error “name torch is not defined” after few steps.
I even tried to reinstall the torch and also tried to restart the runtime. But it won’t work.