|
a |
|
b/docs/readme.md |
|
|
1 |
# Sybil Training Parameters |
|
|
2 |
|
|
|
3 |
### Train Sybil on NLST |
|
|
4 |
|
|
|
5 |
```sh |
|
|
6 |
python train.py \ |
|
|
7 |
--dataset nlst \ |
|
|
8 |
--batch_size 3 \ |
|
|
9 |
--gpus 8 \ |
|
|
10 |
--precision 16 \ |
|
|
11 |
--max_followup 6 \ |
|
|
12 |
--img_file_type png \ |
|
|
13 |
--min_num_images 0 \ |
|
|
14 |
--num_images 200 \ |
|
|
15 |
--use_only_thin_cuts_for_ct \ |
|
|
16 |
--slice_thickness_filter 2.5 \ |
|
|
17 |
--resample_pixel_spacing_prob 0.5 \ |
|
|
18 |
--use_annotations \ |
|
|
19 |
--region_annotations_filepath /path/to/annot_dir/annotations_122020.json \ |
|
|
20 |
--img_mean 128.1722 \ |
|
|
21 |
--img_std 87.1849 \ |
|
|
22 |
--img_size 256 256 \ |
|
|
23 |
--num_chan 3 \ |
|
|
24 |
--limit_train_batches 0.5 \ |
|
|
25 |
--limit_val_batches 1.0 \ |
|
|
26 |
--max_epochs 10 \ |
|
|
27 |
--init_lr 3e-5 \ |
|
|
28 |
--lr_decay 0.1 \ |
|
|
29 |
--weight_decay 1e-2 \ |
|
|
30 |
--momentum 0.9 \ |
|
|
31 |
--dropout 0.1 \ |
|
|
32 |
--optimizer adam \ |
|
|
33 |
--patience 5 \ |
|
|
34 |
--tuning_metric c_index \ |
|
|
35 |
--num_workers 3 \ |
|
|
36 |
--profiler simple \ |
|
|
37 |
--num_sanity_val_steps 0 \ |
|
|
38 |
--save_dir /path/to/model_dir/sybil1 \ |
|
|
39 |
--img_dir /path/to/img_dir/nlst-ct-png \ |
|
|
40 |
--results_path /path/to/results_dir/sybil.results \ |
|
|
41 |
--cache_path /path/to/cache_dir \ |
|
|
42 |
> /path/to/log_dir/sybil.txt 2>&1 |
|
|
43 |
``` |