Diff of /docs/readme.md [000000] .. [d9566e]

Switch to unified view

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
```