Divyanshu Tak
Add BrainIAC IDH Classification app with Vision Transformer model
65bee5d
gpu:
device: cpu
infer:
checkpoints: ./checkpoints/idh_model.ckpt
simclr_checkpoint: ./checkpoints/simclr_vitb.ckpt
threshold: 0.5
image_size: [96, 96, 96]