num_epochs = 30
out_dir = 'out'