Update push_to_hub.py
Browse files- push_to_hub.py +1 -2
push_to_hub.py
CHANGED
|
@@ -4,9 +4,8 @@ from huggingface_hub import PyTorchModelHubMixin
|
|
| 4 |
from model import BERT
|
| 5 |
|
| 6 |
# create model
|
| 7 |
-
config = {"num_channels": 3, "hidden_size": 32, "num_classes": 10}
|
| 8 |
-
model = BERT()
|
| 9 |
|
|
|
|
| 10 |
# save locally
|
| 11 |
model.save_pretrained("genre-pred")
|
| 12 |
|
|
|
|
| 4 |
from model import BERT
|
| 5 |
|
| 6 |
# create model
|
|
|
|
|
|
|
| 7 |
|
| 8 |
+
model = BERT()
|
| 9 |
# save locally
|
| 10 |
model.save_pretrained("genre-pred")
|
| 11 |
|