Rename test-checkpoints.py to test-checkpoint.py
Browse files
test-checkpoints.py → test-checkpoint.py
RENAMED
|
@@ -13,7 +13,7 @@ from train import Transformer, ModelArgs, generate_text # uprav podle názvu so
|
|
| 13 |
# =========================
|
| 14 |
# CONFIG
|
| 15 |
# =========================
|
| 16 |
-
CHECKPOINT_PATH = "checkpoints/
|
| 17 |
TOKENIZER_MODEL_PATH = "tokenizer.model"
|
| 18 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 19 |
|
|
|
|
| 13 |
# =========================
|
| 14 |
# CONFIG
|
| 15 |
# =========================
|
| 16 |
+
CHECKPOINT_PATH = "checkpoints/best.pt"
|
| 17 |
TOKENIZER_MODEL_PATH = "tokenizer.model"
|
| 18 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 19 |
|