You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
585 B
42 lines
585 B
# DATA
|
|
dataset='Tusimple'
|
|
data_root = "Tusimple"
|
|
|
|
# TRAIN
|
|
epoch = 2
|
|
batch_size = 1
|
|
optimizer = 'Adam' #['SGD','Adam']
|
|
# learning_rate = 0.1
|
|
learning_rate = 1e-4
|
|
weight_decay = 1e-4
|
|
momentum = 0.9
|
|
|
|
scheduler = 'cos' #['multi', 'cos']
|
|
# steps = [50,75]
|
|
gamma = 0.1
|
|
warmup = 'linear'
|
|
warmup_iters = 100
|
|
|
|
# NETWORK
|
|
backbone = '18'
|
|
griding_num = 100
|
|
use_aux = False
|
|
|
|
# LOSS
|
|
sim_loss_w = 1.0
|
|
shp_loss_w = 0.0
|
|
|
|
# EXP
|
|
note = 'model'
|
|
|
|
log_path = "./log/"
|
|
|
|
# FINETUNE or RESUME MODEL PATH
|
|
finetune = None
|
|
resume = "ep000.pth"
|
|
|
|
# TEST
|
|
test_model = None
|
|
test_work_dir = None
|
|
|
|
num_lanes = 4 |