File size: 243 Bytes
8af1314 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
---
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)
``` |