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