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