(args)
| 272 | |
| 273 | |
| 274 | def generic_init(args): |
| 275 | if os.path.exists(args.save_dir) == False: |
| 276 | os.makedirs(args.save_dir) |
| 277 | |
| 278 | fmt = '%(asctime)s %(name)s:%(levelname)s: %(message)s' |
| 279 | formatter = logging.Formatter( |
| 280 | fmt, datefmt='%Y-%m-%d %H:%M:%S') |
| 281 | |
| 282 | fh = logging.FileHandler( |
| 283 | '{}/{}_log.txt'.format(args.save_dir, args.save_name), mode='w') |
| 284 | fh.setFormatter(formatter) |
| 285 | |
| 286 | logging.basicConfig(stream=sys.stdout, level=logging.INFO, format=fmt, datefmt='%Y-%m-%d %H:%M:%S') |
| 287 | logger = logging.getLogger() |
| 288 | logger.addHandler(fh) |
| 289 | |
| 290 | logger.info('Arguments') |
| 291 | for arg in vars(args): |
| 292 | logger.info(' {:<22} {}'.format(arg+':', getattr(args,arg)) ) |
| 293 | logger.info('') |
| 294 | |
| 295 | return logger |
| 296 | |
| 297 | |
| 298 | def evaluate(model, criterion, loader, cpu): |
nothing calls this directly
no outgoing calls
no test coverage detected