gpu: device: cpu infer: checkpoints: ./checkpoints/idh_model.ckpt simclr_checkpoint: ./checkpoints/simclr_vitb.ckpt threshold: 0.5 image_size: [96, 96, 96]