diff --git a/time_measure.py b/time_measure.py
index dfdd0a4dda5dada531039f9af955f0f289f02ed6..775057701fbfbafae85d1e4a5a1ccc2bf2f48db1 100644
--- a/time_measure.py
+++ b/time_measure.py
@@ -81,7 +81,7 @@ def inference_time_measurement(model_path, image_path, transform, device = 'cuda
         # Measure memory after prediction
         end_memory = torch.cuda.memory_allocated(device)
         
-        print(f'GPU memory usage: {(end_memory - start_memory)} bytes')
+        print(f'GPU memory usage: {(end_memory - start_memory)/(1024**2)} MB')
         
         
 # Main function
@@ -92,8 +92,8 @@ image_path = get_random_image_path(TEST_FOLDER)
 print(f"Selected image path: {image_path}")
 
 # inputs
-model_path = '/net/travail/bformanek/checkpoints/transfer_checkpoints_resnet18_adam_amp_criterion_balanced/'
-epoch_number = 25
+model_path = '/net/travail/bformanek/checkpoints/transfer_checkpoints_resnet18_augmentation4/'
+epoch_number = 24
 model_name = f'avp_{epoch_number:03d}.pkl'
 model_path = model_path + model_name
 device = 'cuda' # 'cpu' 'cuda'
diff --git a/time_measurement_log.txt b/time_measurement_log.txt
new file mode 100644
index 0000000000000000000000000000000000000000..98b32f8e7003453aa2b831d71e88c43268e53ec0
--- /dev/null
+++ b/time_measurement_log.txt
@@ -0,0 +1,546 @@
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_021_S_7062_MR_ORIG_Sagittal_3D_FLAIR_br_raw_20220329141641279_129_S1118223_I1561596_489.0237_1_121.png
+Inference time masurement using cpu
+Image loaded: 5.86 ms
+Model loaded: 6407.32 ms
+Prediction time for one image: 1039.82 ms
+Compete inference time for one image: 7453.67 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_127_S_4197_MR_Accelerated_Sag_IR-SPGR__br_raw_20110822171753955_196_S119909_I252150_2622.8519_1_121.png
+Inference time masurement using cpu
+Image loaded: 8.11 ms
+Model loaded: 5243.67 ms
+Prediction time for one image: 35.25 ms
+Compete inference time for one image: 5287.45 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/266239_T1_3D_MPRAGE_TRA_9_453.1362_1_121.png
+Inference time masurement using cpu
+Image loaded: 4.07 ms
+Model loaded: 915.58 ms
+Prediction time for one image: 124.35 ms
+Compete inference time for one image: 1044.40 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/264395_2_SEP_DR_MARCHANT_4_1255.1839_0_114.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.50 ms
+Model loaded: 437.10 ms
+Prediction time for one image: 556.97 ms
+Compete inference time for one image: 999.11 ms
+GPU memory usage: 24.25830078125 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/359938_Ax_3D_T1_Gado_8_2515.2712_1_243.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 20.60 ms
+Model loaded: 271.54 ms
+Prediction time for one image: 89.10 ms
+Compete inference time for one image: 381.78 ms
+GPU memory usage: 24.25830078125 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_114_S_1118_MR_MPR__GradWarp__B1_Correction__N3_Br_20070419175300248_S23803_I49767_782.7585_2_076.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 5.00 ms
+Model loaded: 442.89 ms
+Prediction time for one image: 91.36 ms
+Compete inference time for one image: 539.77 ms
+GPU memory usage: 24.25830078125 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/244611_T1_3D_SPACE_FS_SAG_9_295.5908_0_091.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.59 ms
+Model loaded: 1507.36 ms
+Prediction time for one image: 81.10 ms
+Compete inference time for one image: 1593.50 ms
+GPU memory usage: 98.66650390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/199455_Ax_3D_T1_FSPGR_Gado_7_8929.2536_0_243.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.53 ms
+Model loaded: 426.83 ms
+Prediction time for one image: 88.08 ms
+Compete inference time for one image: 519.93 ms
+GPU memory usage: 98.66650390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/309465_T1_TSE_1.0_2min50_701_1495.2168_0_190.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.75 ms
+Model loaded: 380.49 ms
+Prediction time for one image: 85.58 ms
+Compete inference time for one image: 471.29 ms
+GPU memory usage: 98.66650390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/NACC792575_1_T1w_288.8644_1_114.png
+Inference time masurement using cpu
+Image loaded: 4.32 ms
+Model loaded: 2957.29 ms
+Prediction time for one image: 29.15 ms
+Compete inference time for one image: 2991.04 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/197811_Ax_3D_T1_FSPGR_Gado_8_7627.8778_1_243.png
+Inference time masurement using cpu
+Image loaded: 4.30 ms
+Model loaded: 1862.51 ms
+Prediction time for one image: 61.18 ms
+Compete inference time for one image: 1928.27 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/11116_sT1WN_3D_TFE_iso_GADO_4_MN_OFSEP_701_1477.8111_1_114.png
+Inference time masurement using cpu
+Image loaded: 4.19 ms
+Model loaded: 5115.46 ms
+Prediction time for one image: 161.00 ms
+Compete inference time for one image: 5280.96 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_006_S_6696_MR_Sagittal_3D_FLAIR__br_raw_20190402122621272_160_S811841_I1149763_307.0577_0_076.png
+Inference time masurement using cpu
+Image loaded: 4.56 ms
+Model loaded: 16722.84 ms
+Prediction time for one image: 19.89 ms
+Compete inference time for one image: 16747.55 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/267769_SAG_T1_MPRAGE_3D_5001_577.2054_2_152.png
+Inference time masurement using cpu
+Image loaded: 4.52 ms
+Model loaded: 8453.71 ms
+Prediction time for one image: 18.86 ms
+Compete inference time for one image: 8477.24 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/407669_3D_T1_GADO_OK_SENSE_801_real_1667.4131_0_137.png
+Inference time masurement using cpu
+Image loaded: 4.37 ms
+Model loaded: 4719.85 ms
+Prediction time for one image: 14.58 ms
+Compete inference time for one image: 4738.95 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/sub-0095_T1w_3336.6471_0_089.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 5.06 ms
+Model loaded: 352.22 ms
+Prediction time for one image: 68.70 ms
+Compete inference time for one image: 426.29 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC191942_1_FLAIR_837.6159_0_243.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.86 ms
+Model loaded: 374.91 ms
+Prediction time for one image: 74.30 ms
+Compete inference time for one image: 454.43 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_062_S_0768_MR_MPR__GradWarp__B1_Correction__N3__Scaled_Br_20070424122235198_S17527_I50506_561.6648_2_076.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 31.39 ms
+Model loaded: 283.35 ms
+Prediction time for one image: 67.42 ms
+Compete inference time for one image: 382.49 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC700686_1_FLAIR_974.1712_0_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.86 ms
+Model loaded: 360.76 ms
+Prediction time for one image: 67.87 ms
+Compete inference time for one image: 433.79 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T2w/4RTNI_1_S_5101_MR_T2_spc_ns_sag_p2_iso_fs_S5_DIS3D_br_raw_20151207173932880_136_S324928_I572912_331.8526_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.99 ms
+Model loaded: 285.60 ms
+Prediction time for one image: 82.59 ms
+Compete inference time for one image: 373.52 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_051_S_1123_MR_MPRAGE_Repeat__br_raw_20061215133545898_138_S24098_I33308_751.2662_0_076.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.58 ms
+Model loaded: 340.21 ms
+Prediction time for one image: 66.30 ms
+Compete inference time for one image: 411.41 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_007_S_6341_MR_Sagittal_3D_FLAIR__br_raw_20180430155015892_69_S680629_I990575_189.2267_2_121.png
+Inference time masurement using cpu
+Image loaded: 4.06 ms
+cpuModel loaded: 15796.02 ms
+Prediction time for one image: 15.12 ms
+Compete inference time for one image: 15815.35 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC283729_3_FLAIR_348.9785_1_121.png
+Inference time masurement using cpu
+Image loaded: 5.83 ms
+Model loaded: 2936.13 ms
+Prediction time for one image: 20.37 ms
+Compete inference time for one image: 2962.50 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC615802_1_FLAIR_339.4467_0_104.png
+Inference time masurement using cpu
+Image loaded: 4.60 ms
+Model loaded: 5949.78 ms
+Prediction time for one image: 16.75 ms
+Compete inference time for one image: 5971.27 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/OTHER/CuRIOUS_0016_0002_38.4494_1_121.png
+Inference time masurement using cpu
+Image loaded: 4.16 ms
+Model loaded: 3735.86 ms
+Prediction time for one image: 325.72 ms
+Compete inference time for one image: 4066.00 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/264499_3D_TI_EG_CS_1min27_402029_1827.8439_2_228.png
+Inference time masurement using cpu
+Image loaded: 3.90 ms
+Model loaded: 5031.28 ms
+Prediction time for one image: 110.70 ms
+Compete inference time for one image: 5146.19 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIRCE/63402_FLAIR_3D_GADO_TR_5000_14_916.6143_0_076.png
+Inference time masurement using cpu
+Image loaded: 4.15 ms
+Model loaded: 2292.17 ms
+Prediction time for one image: 8.20 ms
+Compete inference time for one image: 2304.81 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_137_S_4816_MR_MPRAGE_br_raw_20120817075456190_168_S160722_I323627_260.8906_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.94 ms
+Model loaded: 293.55 ms
+Prediction time for one image: 146.11 ms
+Compete inference time for one image: 445.07 ms
+GPU memory usage: 14.62255859375 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/265080_T1_3D_MPRAGE_TRA_9_505.1345_0_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.88 ms
+Model loaded: 366.09 ms
+Prediction time for one image: 89.38 ms
+Compete inference time for one image: 460.84 ms
+GPU memory usage: 14.62255859375 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_007_S_0068_MR_MP-RAGE__br_raw_20051201103948809_1_S10355_I9113_1017.5748_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.68 ms
+Model loaded: 272.45 ms
+Prediction time for one image: 88.45 ms
+Compete inference time for one image: 366.01 ms
+GPU memory usage: 14.62255859375 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_127_S_4197_MR_Accelerated_Sag_IR-SPGR__br_raw_20110822171753955_196_S119909_I252150_2622.8519_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 13.34 ms
+Model loaded: 400.26 ms
+Prediction time for one image: 93.78 ms
+Compete inference time for one image: 508.32 ms
+GPU memory usage: 129.4521484375 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIRCE/350177_Sag_CUBE_FLAIR_HS_GADO_9_2021.9937_2_243.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.72 ms
+Model loaded: 309.85 ms
+Prediction time for one image: 79.92 ms
+Compete inference time for one image: 395.46 ms
+GPU memory usage: 129.4521484375 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+^CTraceback (most recent call last):
+  File "/autofs/unityaccount/cremi/bformanek/TRDP_II/./time_measure.py", line 1, in <module>
+    import torch
+  File "/autofs/unityaccount/ens/DeepLearning/python3/tensorflow2/lib/python3.11/site-packages/torch/__init__.py", line 40, in <module>
+    from .torch_version import __version__ as __version__
+  File "<frozen importlib._bootstrap>", line 1178, in _find_and_load
+  File "<frozen importlib._bootstrap>", line 1140, in _find_and_load_unlocked
+  File "<frozen importlib._bootstrap>", line 1080, in _find_spec
+  File "<frozen importlib._bootstrap_external>", line 1504, in find_spec
+  File "<frozen importlib._bootstrap_external>", line 1476, in _get_spec
+  File "<frozen importlib._bootstrap_external>", line 1645, in find_spec
+  File "<frozen importlib._bootstrap_external>", line 161, in _path_isfile
+  File "<frozen importlib._bootstrap_external>", line 153, in _path_is_mode_type
+  File "<frozen importlib._bootstrap_external>", line 147, in _path_stat
+KeyboardInterrupt
+
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC480989_5_FLAIR_189.9778_2_076.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 5.79 ms
+Model loaded: 422.50 ms
+Prediction time for one image: 79.65 ms
+Compete inference time for one image: 508.81 ms
+GPU memory usage: 129.4521484375 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/AIBL_1322_MR_MPRAGE_ADNI_confirmed__br_raw_20141127111835997_38_S237684_I455220_570.9207_1_114.png
+Inference time masurement using cpu
+Image loaded: 4.56 ms
+Model loaded: 7318.06 ms
+Prediction time for one image: 219.64 ms
+Compete inference time for one image: 7542.95 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T2w/NIFD_1_S_0275_MR_T2_spc_ns_sag_p2_iso_fs_S5_DIS3D_br_raw_20160903191223371_55_S496652_I772951_351.4903_2_121.png
+Inference time masurement using cpu
+Image loaded: 4.37 ms
+Model loaded: 6281.07 ms
+Prediction time for one image: 105.92 ms
+Compete inference time for one image: 6392.10 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC971177_1_FLAIR_1012.8245_2_083.png
+Inference time masurement using cpu
+Image loaded: 3.95 ms
+Model loaded: 5830.91 ms
+Prediction time for one image: 164.98 ms
+Compete inference time for one image: 6000.58 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ARWIBO_4292_T1w_2249927.6498_0_054.png
+Inference time masurement using cpu
+Image loaded: 4.39 ms
+Model loaded: 1674.22 ms
+Prediction time for one image: 476.70 ms
+Compete inference time for one image: 2156.11 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/190675_Ax_3D_T1_BRAVO_IR_3_4051.7661_0_243.png
+Inference time masurement using cpu
+Image loaded: 4.26 ms
+Model loaded: 2181.66 ms
+Prediction time for one image: 22.99 ms
+Compete inference time for one image: 2209.57 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T2w/NIFD_1_S_0139_MR_T2_spc_ns_sag_p2_iso_fs_S7_DIS3D_br_raw_20170817090926810_19_S597106_I890346_353.4363_2_121.png
+Inference time masurement using cpu
+Image loaded: 4.27 ms
+Model loaded: 1407.56 ms
+Prediction time for one image: 38.77 ms
+Compete inference time for one image: 1451.39 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_127_S_0112_MR_MPR__GradWarp__B1_Correction_Br_20080318143953240_S11194_I98858_1353.7489_1_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.75 ms
+Model loaded: 354.92 ms
+Prediction time for one image: 241.75 ms
+Compete inference time for one image: 602.36 ms
+GPU memory usage: 46.49853515625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/397526_SAG_T1_RECO_203_real_1771.1317_0_066.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.62 ms
+Model loaded: 332.73 ms
+Prediction time for one image: 91.68 ms
+Compete inference time for one image: 429.92 ms
+GPU memory usage: 46.49853515625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/198607_T1MPRAGE_sag_iso_1.0_8_425.5113_0_091.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 17.94 ms
+Model loaded: 288.11 ms
+Prediction time for one image: 91.58 ms
+Compete inference time for one image: 398.55 ms
+GPU memory usage: 46.49853515625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/36380_SAG_3D_MPRAGE_WE_0.8_ISO_HR_9_566.0449_1_274.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 5.02 ms
+Model loaded: 941.73 ms
+Prediction time for one image: 68.73 ms
+Compete inference time for one image: 1015.80 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/375937_WIP_T1_3D_TFE_3MIN_201_925877.7138_1_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.66 ms
+Model loaded: 260.26 ms
+Prediction time for one image: 68.17 ms
+Compete inference time for one image: 333.43 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_052_S_6832_MR_Sagittal_3D_FLAIR__br_raw_20191120161519593_131_S898885_I1258695_154.6390_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.62 ms
+Model loaded: 313.60 ms
+Prediction time for one image: 70.59 ms
+Compete inference time for one image: 389.12 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_002_S_4225_MR_MPRAGE_SENSE2__br_raw_20110921132732496_125_S122885_I257274_459.5388_2_121.png
+Inference time masurement using cpu
+Image loaded: 4.04 ms
+Model loaded: 1875.68 ms
+Prediction time for one image: 14.89 ms
+Compete inference time for one image: 1894.76 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/26100_t1_spc_ns_sag_p2_9_434.3826_0_083.png
+Inference time masurement using cpu
+Image loaded: 3.96 ms
+Model loaded: 2401.05 ms
+Prediction time for one image: 15.93 ms
+Compete inference time for one image: 2421.10 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/167659_T1_3D_MPRAGE_TRA_5_377.7154_1_121.png
+Inference time masurement using cpu
+Image loaded: 4.12 ms
+Model loaded: 1355.87 ms
+Prediction time for one image: 17.09 ms
+Compete inference time for one image: 1377.24 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIRCE/81146_3D_FLAIR_5000_GADO_801_6779.3233_2_243.png
+Inference time masurement using cpu
+Image loaded: 4.02 ms
+Model loaded: 30653.38 ms
+Prediction time for one image: 16.88 ms
+Compete inference time for one image: 30674.43 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC752141_1_FLAIR_186.5808_0_112.png
+Inference time masurement using cpu
+Image loaded: 3.97 ms
+Model loaded: 375.37 ms
+Prediction time for one image: 14.88 ms
+Compete inference time for one image: 394.37 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/260980_SAGITTAL_T1_eg3D_0.9_5_334.7883_0_091.png
+Inference time masurement using cpu
+Image loaded: 4.18 ms
+Model loaded: 1402.22 ms
+Prediction time for one image: 16.09 ms
+Compete inference time for one image: 1422.64 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/264953_SAG_T1_SPACE_iso_GADO_17_650.3921_0_091.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.64 ms
+Model loaded: 287.13 ms
+Prediction time for one image: 72.03 ms
+Compete inference time for one image: 364.15 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIRCE/385070_FLAIR_3D_GADO_TR_5000_7_691.2526_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.77 ms
+Model loaded: 355.07 ms
+Prediction time for one image: 68.50 ms
+Compete inference time for one image: 428.68 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_153_S_4125_MR_MPRAGE_GRAPPA2_br_raw_20110720111302542_133_S115442_I245816_531.2860_0_083.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 5.11 ms
+Model loaded: 313.10 ms
+Prediction time for one image: 67.83 ms
+Compete inference time for one image: 386.36 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_114_S_0979_MR_MPR-R__GradWarp__B1_Correction_Br_20070213202520174_S21934_I39864_763.6590_0_091.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.63 ms
+Model loaded: 881.35 ms
+Prediction time for one image: 68.18 ms
+Compete inference time for one image: 954.47 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/DLBS_0028596_s1_a1_1851.5627_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 5.11 ms
+Model loaded: 261.38 ms
+Prediction time for one image: 77.17 ms
+Compete inference time for one image: 344.00 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_007_S_6341_MR_Sagittal_3D_FLAIR__br_raw_20180430155015892_69_S680629_I990575_189.2267_0_076.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.52 ms
+Model loaded: 285.53 ms
+Prediction time for one image: 70.44 ms
+Compete inference time for one image: 360.83 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/162912_Ax_3D_T1_Gado_7_3540.1661_0_243.png
+Inference time masurement using cpu
+Image loaded: 4.01 ms
+Model loaded: 2973.95 ms
+Prediction time for one image: 13.90 ms
+Compete inference time for one image: 2992.01 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC525519_1_FLAIR_169.0394_0_112.png
+Inference time masurement using cpu
+Image loaded: 4.34 ms
+Model loaded: 1745.36 ms
+Prediction time for one image: 19.62 ms
+Compete inference time for one image: 1769.46 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_109_S_6220_MR_Sagittal_3D_FLAIR__br_raw_20180411153321880_114_S675565_I984321_123.8017_2_121.png
+Inference time masurement using cpu
+Image loaded: 4.70 ms
+Model loaded: 6259.79 ms
+Prediction time for one image: 15.67 ms
+Compete inference time for one image: 6280.30 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/ADNI_116_S_6543_MR_Sagittal_3D_FLAIR__br_raw_20180821112003137_121_S719357_I1038944_282.1697_1_121.png
+Inference time masurement using cpu
+Image loaded: 4.27 ms
+Model loaded: 13141.84 ms
+Prediction time for one image: 45.38 ms
+Compete inference time for one image: 13191.76 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/DLBS_0028637_s1_a1_1635.0307_2_121.png
+Inference time masurement using cpu
+Image loaded: 3.81 ms
+Model loaded: 3873.48 ms
+Prediction time for one image: 15.17 ms
+Compete inference time for one image: 3892.62 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/261941_T1_SPACE_3D_6_287.6819_2_121.png
+Inference time masurement using cpu
+Image loaded: 5.76 ms
+Model loaded: 13807.26 ms
+Prediction time for one image: 68.10 ms
+Compete inference time for one image: 13881.26 ms
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1wCE/361459_3D_SAG_T1_GEIR_6_719.8513_0_100.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 18.39 ms
+Model loaded: 297.39 ms
+Prediction time for one image: 74.24 ms
+Compete inference time for one image: 390.35 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/T1w/ADNI_130_S_4352_MR_MPRAGE_br_raw_20111121102623843_152_S130237_I267711_460.8166_2_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.58 ms
+Model loaded: 355.61 ms
+Prediction time for one image: 68.78 ms
+Compete inference time for one image: 429.29 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ python3 ./time_measure.py 
+Selected image path: /net/travail/bformanek/MRI_dataset/test/FLAIR/NACC874587_1_FLAIR_502.7401_1_121.png
+cuda is available
+Inference time masurement using cuda
+Image loaded: 4.85 ms
+Model loaded: 271.77 ms
+Prediction time for one image: 68.35 ms
+Compete inference time for one image: 345.29 ms
+GPU memory usage: 51.400390625 MB
+(tensorflow2) bformanek@falcon:~/TRDP_II$ 
\ No newline at end of file
diff --git a/train_results.txt b/train_results.txt
index 7293f411098e3ece1bdbb743b2b6c37e32fe9625..2a59a4859d6679d703de4bd8a2f41591bc30c29d 100644
--- a/train_results.txt
+++ b/train_results.txt
@@ -133,12 +133,6 @@ RandomResample(scale_factor=4),
         size: 43MB
     checkpoints: /net/travail/bformanek/checkpoints/transfer_checkpoints_resnet18_augmentation1
     note: poor results on validation and test sets
-        - confusion matrix:
-            t2 -> t2* small confusion (139)
-            flair -> t2* huge confusion (993)
-            t1 -> t2* large confusion (221)
-            t1ce -> t2* small confusion (81)
-            t1 -> t2 medium (128)
 
 augmentation2: lower random contrast, blur and resample 
 MODEL_NAME = 'resnet18'