diff --git a/Configuration/config.ini b/Configuration/config.ini
index ae20d94f071f1947cc7890966573c12bbcd2d9ba..fedb31786ee8e94c9842dda2a24209c43827183e 100644
--- a/Configuration/config.ini
+++ b/Configuration/config.ini
@@ -12,7 +12,7 @@ max_atoms_in_one_type=324
 dim_encoder = 768
 
 [MODEL_LINKER]
-nhead=8
+nhead=16
 dim_emb_atom = 256
 dim_feedforward_transformer = 512
 num_layers=3
@@ -24,7 +24,7 @@ dropout=0.1
 sinkhorn_iters=5
 
 [MODEL_TRAINING]
-batch_size=32
+batch_size=16
 epoch=30
 seed_val=42
 learning_rate=2e-3
\ No newline at end of file
diff --git a/logs/logs/Accuracy_Train/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.1 b/logs/logs/Accuracy_Train/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.1
new file mode 100644
index 0000000000000000000000000000000000000000..836da89cfb7f563cf1ff1108b2eed3dd66ef9ff8
Binary files /dev/null and b/logs/logs/Accuracy_Train/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.1 differ
diff --git a/logs/logs/Accuracy_Validation/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.3 b/logs/logs/Accuracy_Validation/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.3
new file mode 100644
index 0000000000000000000000000000000000000000..1945341e16f8ba0d0acfb9bd7fa891e2b53b9552
Binary files /dev/null and b/logs/logs/Accuracy_Validation/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.3 differ
diff --git a/logs/logs/Loss_Train/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.2 b/logs/logs/Loss_Train/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.2
new file mode 100644
index 0000000000000000000000000000000000000000..89acad93d838280f7f49a8320892c5c3d1b5440c
Binary files /dev/null and b/logs/logs/Loss_Train/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.2 differ
diff --git a/logs/logs/Loss_Validation/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.4 b/logs/logs/Loss_Validation/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.4
new file mode 100644
index 0000000000000000000000000000000000000000..27f3fb8fb1adb2f8cb3f308002e5aceb72ce8ff3
Binary files /dev/null and b/logs/logs/Loss_Validation/events.out.tfevents.1656000220.co2-slurm-ngrtx01.97600.4 differ
diff --git a/logs/logs/events.out.tfevents.1655998955.co2-slurm-ngrtx01.97600.0 b/logs/logs/events.out.tfevents.1655998955.co2-slurm-ngrtx01.97600.0
new file mode 100644
index 0000000000000000000000000000000000000000..b701efbbdba2cec059713e10b820c830a690ea70
Binary files /dev/null and b/logs/logs/events.out.tfevents.1655998955.co2-slurm-ngrtx01.97600.0 differ