SDGPA / rain2 /exp_args.py
roujin's picture
init
44a8b4b verified
raw
history blame contribute delete
805 Bytes
class Args:
synth_path = '/data/shared/luojun/rain_intermediate/leftImg8bit/train/'
source_path = '/data/shared/luojun/rain_target'
target_type = 'acdc_rain'
target_path = '/data/shared/luojun/ACDC'
experiment_name = '/data/shared/luojun/runs/rain2'
stop_epoch = 65
debug = False
load = '/data/shared/luojun/runs/rain1/weights/checkpoint.pth.tar'
resume = None
epochs = 100
seed = 42
sim_weight = 0.0
model = 'resnet-50'
ssl = 'sim'
optimizer = 'sgd'
momentum = 0.9
wd = 0.0001
schedular = 'poly'
batch_size = 4
lr_head = 0.001
lr = 0.001
target_size = (512, 1024)
crop = 'both_random'
crop_size = (384, 768)
blur = False
cutout = False
jitter = 0.5
scale = 0.0
xs = False
workers = 4