Spaces:
Sleeping
Sleeping
File size: 166 Bytes
65bee5d |
1 2 3 4 5 6 7 |
gpu:
device: cpu
infer:
checkpoints: ./checkpoints/idh_model.ckpt
simclr_checkpoint: ./checkpoints/simclr_vitb.ckpt
threshold: 0.5
image_size: [96, 96, 96] |