EndoSAM / config /finetune.yaml
Chris Xiao
init model
2df812d
raw
history blame contribute delete
783 Bytes
experiment_name: EndoSAM
model:
class_num: 2
sam_model_customized: true
sam_model_type: default
sam_model_dir: /home/iu/Desktop/EndoSAM/sam_ckpts/default.pth
encoder_size: 1024
dataset:
class_names:
- background
- instrument-clapster
dataset_dir: /home/iu/Downloads/all_data
img_format: jpg
ann_format: png
losses:
ce:
weight: 0.5
mse:
weight: 0.5
opt_params:
lr_default: 0.0001
max_iter: 1
val_iter: 5
train_bs: 2
val_bs: 2
test_bs: 1
num_workers: 20
num_token: 4
log_folder: /home/iu/Desktop/EndoSAM_experiment/log
model_folder: /home/iu/Desktop/EndoSAM_experiment/model
ckpt_folder: /home/iu/Desktop/EndoSAM_experiment/checkpoint
plot_folder: /home/iu/Desktop/EndoSAM_experiment/plots
test_folder: /home/iu/Desktop/EndoSAM_experiment/inference