README.md

metadata

license: mit

see https://github.com/ZreXoc/nailong-classification

model saved with:

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