local_cpu_time_measurement_log.txt

File "/autofs/unityaccount/cremi/bformanek/TRDP_II/time_measure.py", line 68, in inference_time_measurement
    model = torch.load(model_path, map_location=torch.device('cpu'), weights_only=True)
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/lib/python3/dist-packages/torch/serialization.py", line 788, in load
    raise pickle.UnpicklingError(UNSAFE_MESSAGE + str(e)) from None
_pickle.UnpicklingError: Weights only load failed. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution.Do it only if you get the file from a trusted source. WeightsUnpickler error: Unsupported class timm.models.mobilenetv3.MobileNetV3

cuda is not available

(.venv) bformanek@falcon:~/TRDP_II$ python time_measure.py 
MEASUREMENT - 1
Selected image path: /net/travail/bformanek/MRI_dataset/test/T2w/IXI222-Guys-0819-T2_358.1985_1_121.png
Copied to local image path: /net/cremi/bformanek/TRDP_II/local_images/IXI222-Guys-0819-T2_358.1985_1_121.png
MODEL: 
transfer_checkpoints_mobilenetv3_small_100.lamb_in1k_avp_028.pkl
DEVICE: cpu
Inference time masurement using cpu
Image loaded: 4.49 ms
Model loaded: 3728.52 ms
Prediction time for one image: 398.29 ms
Compete inference time for one image: 4131.33 ms
DEVICE: cuda
cuda is not available
MODEL: 
transfer_checkpoints_efficientnet_b0_adam_amp_criterion_balanced_avp_026.pkl
DEVICE: cpu
Inference time masurement using cpu
Image loaded: 2.10 ms
Model loaded: 33.05 ms
Prediction time for one image: 642.03 ms
Compete inference time for one image: 677.21 ms
DEVICE: cuda
cuda is not available
MODEL: 
transfer_checkpoints_resnet18.fb_ssl_yfcc100m_ft_in1k_avp_022.pkl
DEVICE: cpu
Inference time masurement using cpu
Image loaded: 53.70 ms
Model loaded: 19.95 ms
Prediction time for one image: 17.19 ms
Compete inference time for one image: 90.88 ms
DEVICE: cuda
cuda is not available
MODEL: 
transfer_checkpoints_mobilenetv4_conv_large.e500_r256_in1k_avp_004.pkl
DEVICE: cpu
Inference time masurement using cpu
Image loaded: 1.74 ms
Model loaded: 62.56 ms
Prediction time for one image: 53.74 ms
Compete inference time for one image: 118.08 ms
DEVICE: cuda
cuda is not available
MODEL: 
transfer_checkpoints_resnet18_adam_amp_criterion_balanced_avp_025.pkl
DEVICE: cpu
Inference time masurement using cpu
Image loaded: 77.87 ms
Model loaded: 18.53 ms
Prediction time for one image: 34.57 ms
Compete inference time for one image: 131.02 ms
DEVICE: cuda
cuda is not available
MODEL: 
transfer_checkpoints_resnet18_augmentation2_avp_019.pkl
DEVICE: cpu
Inference time masurement using cpu
Image loaded: 19.00 ms
Model loaded: 24.10 ms
Prediction time for one image: 9.96 ms