Initial upload
Browse files- config.json +1 -1
- pytorch_model.bin +1 -1
config.json
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
{"SEQ_LEN": 64, "BATCH_SIZE": 8, "EPOCHS":
|
|
|
|
| 1 |
+
{"SEQ_LEN": 64, "BATCH_SIZE": 8, "EPOCHS": 4, "EMBED_DIM": 128, "NUM_HEADS": 4, "NUM_LAYERS": 2, "LR": 0.0003, "TOKENIZER_NAME": "gpt2", "DEVICE": "cuda"}
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 53828454
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2b87b8467f92396481f611947bc1d975c476b8812f0298cb0a0732529460f5b5
|
| 3 |
size 53828454
|