Update training scripts

Author: ikergarcia1996

training_scripts/GPU/TEDD_1140_base.sh

@@ -5,7 +5,7 @@ python3 train.py --train_new \
   --encoder_type transformer \
   --batch_size 16 \
   --accumulation_steps 4 \
-  --max_epochs 12 \
+  --max_epochs 40 \
   --cnn_model_name efficientnet_b4 \
   --num_layers_encoder 4 \
   --embedded_size 512 \

training_scripts/GPU/TEDD_1140_large.sh

@@ -5,8 +5,8 @@ python3 train.py --train_new \
   --encoder_type transformer \
   --batch_size 16 \
   --accumulation_steps 4 \
-  --max_epochs 12 \
-  --cnn_model_name efficientnet_v2_l \
+  --max_epochs 40 \
+  --cnn_model_name efficientnet_b7 \
   --num_layers_encoder 6 \
   --embedded_size 512 \
   --learning_rate 1e-5 \

training_scripts/TPU/TEDD_1140_base.sh

@@ -5,9 +5,10 @@ python3 train.py --train_new \
   --val_dir  ../gtaai_datasets/dev \
   --output_dir models/tedd_1104_base \
   --encoder_type transformer \
+  --dataloader_num_workers 32 \
   --batch_size 8 \
   --accumulation_steps 1 \
-  --max_epochs 12 \
+  --max_epochs 40 \
   --cnn_model_name efficientnet_b4 \
   --num_layers_encoder 4 \
   --embedded_size 512 \

training_scripts/TPU/TEDD_1140_large.sh

@@ -5,10 +5,11 @@ python3 train.py --train_new \
   --val_dir  ../gtaai_datasets/dev \
   --output_dir models/tedd_1104_large \
   --encoder_type transformer \
+  --dataloader_num_workers 32 \
   --batch_size 8 \
   --accumulation_steps 1 \
-  --max_epochs 12 \
-  --cnn_model_name efficientnet_v2_l \
+  --max_epochs 40 \
+  --cnn_model_name efficientnet_b7 \
   --num_layers_encoder 6 \
   --embedded_size 512 \
   --learning_rate 1e-5 \