| 12345678 |
- def save_parameters(args, run_name):
- with open(os.path.join(args.log_path, run_name)+'/parameters.txt', 'w') as f:
- f.write('num_blocks {}, lr {}, beta1 {} beta2 {}, batch_size {} gamma {} scheduler_step {}'.format(
- args.num_blocks, args.lr, args.beta1, args.beta2, args.batch_size,
- args.gamma, args.scheduler_step
- ))
|
