model saved with:

checkpoint = {'model': model.state_dict(), 'optimizer': optimizer.state_dict(), \
      'epoch': epoch}
torch.save(checkpoint , path)

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support