PTI/configs/global_config.py

# 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 = ''