Aadish Sharma
Aadish Sharma
I am also facing the same issue with v100
these solutions not working for me
Traceback (most recent call last): File "alpaca_fine_tuning.py", line 259, in trainer.train() File "/home/kokil/.local/lib/python3.8/site-packages/transformers/trainer.py", line 1645, in train return inner_training_loop( File "/home/kokil/.local/lib/python3.8/site-packages/transformers/trainer.py", line 1938, in _inner_training_loop tr_loss_step = self.training_step(model, inputs) File...
with torch.autocast("cuda"): trainer.train() @xllau do you mean like this?