(args)
| 50 | |
| 51 | """checking arguments""" |
| 52 | def check_args(args): |
| 53 | # --checkpoint_dir |
| 54 | check_folder(args.checkpoint_dir) |
| 55 | |
| 56 | # --result_dir |
| 57 | check_folder(args.result_dir) |
| 58 | |
| 59 | # --result_dir |
| 60 | check_folder(args.log_dir) |
| 61 | |
| 62 | # --sample_dir |
| 63 | check_folder(args.sample_dir) |
| 64 | |
| 65 | # --epoch |
| 66 | try: |
| 67 | assert args.epoch >= 1 |
| 68 | except: |
| 69 | print('number of epochs must be larger than or equal to one') |
| 70 | |
| 71 | # --batch_size |
| 72 | try: |
| 73 | assert args.batch_size >= 1 |
| 74 | except: |
| 75 | print('batch size must be larger than or equal to one') |
| 76 | return args |
| 77 | |
| 78 | |
| 79 | """main""" |
no test coverage detected