File size: 224 Bytes
bb0f5a9
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
# Device
cuda_visible_devices = '0'
device = 'cuda:0'

# Logs
training_step = 1
image_rec_result_log_snapshot = 100
pivotal_training_steps = 0
model_snapshot_interval = 400

# Run name to be updated during PTI
run_name = ''