Medical-SAM-Adapter
Medical-SAM-Adapter copied to clipboard
run train.py
Hello author, I ran the train.py file with an error. The following is the specific content of the error:
What I entered: python train.py -net sam -mod sam_adpt -exp_name medical-sam-ada -sam_ckpt checkpoint/sam/sam_vit_b_01ec64.pth -image_size 1024 -b 8 -dataset isic --data_path ISBI2016_dataset
Namespace(net='sam', baseline='unet', seg_net='transunet', mod='sam_adpt', exp_name='medical-sam-ada', type='map', vis=None, reverse=False, pretrain=False, val_freq=100, gpu=True, gpu_device=0, sim_gpu=0, epoch_ini=1, image_size=1024, out_size=256, patch_size=2, dim=512, depth=1, heads=16, mlp_dim=1024, w=4, b=8, s=True, warm=1, lr=0.0001, uinch=1, imp_lr=0.0003, weights=0, base_weights=0, sim_weights=0, distributed='none', dataset='isic', sam_ckpt='checkpoint/sam/sam_vit_b_01ec64.pth', thd=False, chunk=96, num_sample=4, roi_size=96, evl_chunk=None, data_path='ISBI2016_dataset', path_helper={'prefix': 'logs\medical-sam-ada_2023_11_18_10_37_15', 'ckpt_path': 'logs\medical-sam-ada_2023_11_18_10_37_15\Model', 'log_path': 'logs\medical-sam-ada_2023_11_18_10_37_15\Log', 'sample_path': 'logs\medical-sam-ada_2023_11_18_10_37_15\Samples'})
n