예제 #1
0
 def setUpClass(cls):
     out = run_validation(**{'--use-synthetic-data': True,
                             '--epochs': 2,
                             '--seed': 3,
                             '--learning-rate': '0.1',
                             '--batches-per-step': 160})
     cls.out = out
     cls.infer = get_log(out)
예제 #2
0
 def setUpClass(cls):
     out = run_train(**{'--use-synthetic-data': True,
                        '--epochs': 1,
                        '--seed': 3,
                        '--learning-rate': 0.1,
                        '--batches-per-step': 1600})
     cls.out = out
     cls.training = get_log(out)