Skip to content
Snippets Groups Projects
Select Git revision
  • e54dabf99c2e02ea7b0b4e49500510677333dcc4
  • main default protected
2 results

local_cpu_time_measurement_log.txt

Blame
  • local_cpu_time_measurement_log.txt 24.25 KiB
    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