Create pytorch_model.bin
Browse files- pytorch_model.bin +2 -0
pytorch_model.bin
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Saving the model's weights
|
| 2 |
+
torch.save(model.state_dict(), 'pytorch_model.bin')
|