(args)
| 122 | |
| 123 | |
| 124 | def init_args(args): |
| 125 | args.steps_per_epoch = 10000000000 |
| 126 | args.data_dir = args.data_file+args.data_dir |
| 127 | os.environ['CUDA_VISIBLE_DEVICS'] = args.gpu_id |
| 128 | os.makedirs(args.output, exist_ok=True) |
| 129 | sys.stdout = Tee(os.path.join(args.output, 'out.txt')) |
| 130 | sys.stderr = Tee(os.path.join(args.output, 'err.txt')) |
| 131 | args = act_param_init(args) |
| 132 | return args |
no test coverage detected