| 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) | |
| ``` |
| 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) | |
| ``` |