-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCL_train.sh
71 lines (60 loc) · 1.36 KB
/
CL_train.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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#!/usr/bin/env bash
set -x
# Define variables
PUS_PER_NODE=4
BATCH_SIZE=8
MODEL_NAME="dn_detr"
PREFETCH_SIZE=2
COCO_PATH=
OUTPUT_DIR=
START_TASK=
START_EPOCH=
TASK_EPOCHS=30
NUM_WORKERS=12
TOTAL_CLASSES=59
TEST_FILE_LIST="did pz ve"
PRETRAINED_MODEL=
# replay method
LIMIT_IMAGE=1200
LEAST_IMAGE=12 # 1%
TASK=3
REHEARSAL_FILE=
SAMPLING_STRATEGY="hierarchical"
SAMPLING_MODE="GM"
# Fisher
FISHER_MODEL=
CER_MODE="fisher"
# Prepare the command
CMD="PUS_PER_NODE=4 ./tools/run_dist_launch.sh $PUS_PER_NODE ./configs/r50_dn_detr.sh \
--batch_size $BATCH_SIZE \
--prefetch $PREFETCH_SIZE \
--model_name $MODEL_NAME \
--coco_path $COCO_PATH \
--output_dir $OUTPUT_DIR \
--test_file_list $TEST_FILE_LIST \
--orgcocopath \
--use_dn \
--Branch_Incremental \
--Total_Classes $TOTAL_CLASSES \
--Total_Classes_Names \
--Task_Epochs $TASK_EPOCHS \
--start_task $START_TASK \
--start_epoch $START_EPOCH \
--num_workers $NUM_WORKERS \
--Task $TASK \
--verbose \
--LG \
--Rehearsal_file $REHEARSAL_FILE \
--Rehearsal \
--limit_image $LIMIT_IMAGE \
--least_image $LEAST_IMAGE \
--Sampling_strategy $SAMPLING_STRATEGY \
--Sampling_mode $SAMPLING_MODE \
--AugReplay \
--CER $CER_MODE \
--fisher_model $FISHER_MODEL \
$@"
# Print the command
echo $CMD
# Run the command
eval $CMD