-
Notifications
You must be signed in to change notification settings - Fork 2
/
run_tsd.sh
45 lines (41 loc) · 1.3 KB
/
run_tsd.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
OUTPUT_DIR='logs/occ_duke/tsd_5'
# # python3 train_tsd.py \
# # --config_file configs/OCC_Duke/vit_base.yml \
# # INPUT.MASK_PREPROCESS eight \
# # MODEL.NUM_PARTS 8 \
# # SOLVER.BASE_LR 0.004 \
# # MODEL.DEVICE_ID "('0')" \
# # MODEL.WITH_FORE_HEAD True \
# # MODEL.WITH_DIVERSE_LOSS True \
# # MODEL.PRETRAIN_PATH logs/pretrain_model/jx_vit_base_p16_224-80ecf9dd.pth \
# # OUTPUT_DIR $OUTPUT_DIR
# python3 test_tsd.py \
# --config_file configs/OCC_Duke/vit_base.yml \
# TEST.WEIGHT $OUTPUT_DIR/transformer_90.pth \
# INPUT.MASK_PREPROCESS eight \
# MODEL.NUM_PARTS 8 \
# MODEL.DEVICE_ID "('0')" \
# MODEL.WITH_FORE_HEAD True \
# DATASETS.NAMES new_occ_duke \
# TEST.NEW_METRIC True \
# OUTPUT_DIR $OUTPUT_DIR
python3 train_tsd.py \
--config_file configs/OCC_Duke/vit_base.yml \
INPUT.MASK_PREPROCESS five \
MODEL.NUM_PARTS 5 \
SOLVER.BASE_LR 0.004 \
MODEL.DEVICE_ID "('0')" \
MODEL.WITH_FORE_HEAD True \
MODEL.WITH_DIVERSE_LOSS True \
MODEL.PRETRAIN_PATH logs/pretrain_model/jx_vit_base_p16_224-80ecf9dd.pth \
OUTPUT_DIR $OUTPUT_DIR
python3 test_tsd.py \
--config_file configs/OCC_Duke/vit_base.yml \
TEST.WEIGHT $OUTPUT_DIR/transformer_90.pth \
INPUT.MASK_PREPROCESS five \
MODEL.NUM_PARTS 5 \
MODEL.DEVICE_ID "('0')" \
MODEL.WITH_FORE_HEAD True \
DATASETS.NAMES new_occ_duke \
TEST.NEW_METRIC True \
OUTPUT_DIR $OUTPUT_DIR