-
Notifications
You must be signed in to change notification settings - Fork 2
/
train_chexpert.sh
24 lines (23 loc) · 1.23 KB
/
train_chexpert.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
python3 -W ignore main_chexpert.py --wandb_project_name debug \
--log_dir /data/selfsupervision/log/chexpert \
--num_epoch 3 \
--model_name densenet121 \
--optimizer adam \
--lr 0.001 \
--batch_size 16 \
--num_workers 8 \
--iters_per_eval 100 \
--gpu_ids 3 \
--resize_shape 320 \
--crop_shape 320 \
--rotation_range 20 \
--img_type Frontal \
--lr_decay 0.1 \
--weight_decay 0.0 \
--momentum 0.9 \
--sgd_dampening 0.0 \
--uncertain ignore \
--threshold 0.5 \
--gaussian_noise_mean 0.0 \
--gaussian_noise_std 0.05 \
--gaussian_blur 1.0