Update get_validation_data

QueensGambit · Aug 5, 2024 · 8b5b1d2 · 8b5b1d2
1 parent aafca5c
commit 8b5b1d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/DeepCrazyhouse/src/training/train_cli.py b/DeepCrazyhouse/src/training/train_cli.py
@@ -81,7 +81,7 @@ def main():
 
     update_train_config_via_args(args, train_config)
 
-    val_data, x_val, _ = get_validation_data(train_config)
+    val_data, x_val = get_validation_data(train_config)
     input_shape = x_val[0].shape
     fill_train_config(train_config, x_val)
 

diff --git a/DeepCrazyhouse/src/training/train_cli_util.py b/DeepCrazyhouse/src/training/train_cli_util.py
@@ -284,7 +284,7 @@ def get_validation_data(train_config: TrainConfig):
     """
     pgn_dataset_arrays_dict = load_pgn_dataset(dataset_type='val', part_id=0, verbose=True, normalize=train_config.normalize)
     val_data = get_data_loader(pgn_dataset_arrays_dict, train_config, shuffle=False)
-    return val_data, x_val, yp_val
+    return val_data, pgn_dataset_arrays_dict["x"]
 
 
 def print_model_summary(input_shape: tuple, model, x_val) -> None: