long-text-token-classification
long-text-token-classification copied to clipboard
How to train with multiple gpus?
It will raise an error when I tried to train tez model on multiple gpus.
model = nn.DataParallel(model, device_ids=[0,1,2,3])
AttributeError: 'DataParallel' object has no attribute 'fit'
How can I do this? Thank you so much
dataparallel doesnt work with Tez for now
Are there any ways?
write your own training/evaluation loop
OK, thank you
OK, thank you “write your own training/evaluation loop” how can i do this loop?thank you so much!