pytorch-autocomplete

pytorch-autocomplete copied to clipboard

UnboundLocalError: local variable 'loss' referenced before assignment

1

Hi, When I run the train function I get this error. File "C:\Desktop\work\sanstitre1.py", line 256, in train train_history['loss'].append(loss.item()) UnboundLocalError: local variable 'loss' referenced before assignment Any help?? Imen BA

Imenbaa

training is slow, is it possible to add multi-gpu ?

3

Hi, Hope you are all well ! I tried to train pytorch autocomplete on a 200Mb file and it is quiet slow. Is it possible to add the dataparallel function...

ghost