nailong_model / README.md
XiC1's picture
Update README.md
8af1314 verified
---
license: mit
---
see https://github.com/ZreXoc/nailong-classification
model saved with:
```python
checkpoint = {'model': model.state_dict(), 'optimizer': optimizer.state_dict(), \
'epoch': epoch}
torch.save(checkpoint , path)
```