From 624551130ebc140e51258cdfe4786bc6c65046e5 Mon Sep 17 00:00:00 2001
From: LVeefkind <larsveefkind@gmail.com>
Date: Wed, 20 Nov 2024 14:23:22 +0100
Subject: [PATCH] add pos embedding to run name

---
 neural_networks/train_nn.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/neural_networks/train_nn.py b/neural_networks/train_nn.py
index e00c042..df9db03 100644
--- 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)