flpelerin commited on
Commit ·
00efcbd
1
Parent(s): aed0c2d
fix
Browse files
train.py
CHANGED
|
@@ -31,19 +31,20 @@ print(f"Tokenzier has {vocab_size} unique tokens")
|
|
| 31 |
|
| 32 |
# Initialize W&B Logging
|
| 33 |
wandb.init(project="minGRU-Training", config={
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
|
|
|
| 45 |
settings=wandb.Settings(api_key="860f8753998c6e6dc356914de07e8855aa2f9642")
|
| 46 |
-
|
| 47 |
|
| 48 |
# ============================
|
| 49 |
# Load and Preprocess Dataset
|
|
|
|
| 31 |
|
| 32 |
# Initialize W&B Logging
|
| 33 |
wandb.init(project="minGRU-Training", config={
|
| 34 |
+
"dataset_path": dataset_path,
|
| 35 |
+
"num_epochs": num_epochs,
|
| 36 |
+
"batch_size": batch_size,
|
| 37 |
+
"seq_length": seq_length,
|
| 38 |
+
"learning_rate": learning_rate,
|
| 39 |
+
"infer_step": infer_step,
|
| 40 |
+
"input_len": input_len,
|
| 41 |
+
"num_predict": num_predict,
|
| 42 |
+
"reset_state_every": reset_state_every,
|
| 43 |
+
"validate_every": validate_every,
|
| 44 |
+
"device": str(device)
|
| 45 |
+
},
|
| 46 |
settings=wandb.Settings(api_key="860f8753998c6e6dc356914de07e8855aa2f9642")
|
| 47 |
+
)
|
| 48 |
|
| 49 |
# ============================
|
| 50 |
# Load and Preprocess Dataset
|