add pos embedding to run name

jurjen93 · Nov 20, 2024 · 6245511 · 6245511
1 parent a71f66f
commit 6245511
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neural_networks/train_nn.py b/neural_networks/train_nn.py
@@ -424,6 +424,7 @@ def main(
         alpha=alpha,
         lift=lift,
         flip_augmentations=flip_augmentations,
+        tune_pos_embed=tune_pos_embed,
     )
 
     writer = get_tensorboard_logger(logging_dir)