Update README.md

README.md

@@ -1,3 +1,14 @@
----
-license: mit
----
+---
+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)
+
+```