Ejemplo n.º 1
0
 def test_visualize(self):
     model_dir = os.path.abspath('../data/model/20160620-173927')
     create_checkpoint_file(model_dir, 'model.ckpt-500000')
     argv = [model_dir, 
             '--model_def', 'models.nn4' ]
     args = visualize.parse_arguments(argv)
     visualize.main(args)
Ejemplo n.º 2
0
 def test_visualize(self):
     model_dir = os.path.abspath('../data/model/20160620-173927')
     create_checkpoint_file(model_dir, 'model.ckpt-500000')
     argv = [model_dir, 
             '--model_def', 'models.nn4' ]
     args = visualize.parse_arguments(argv)
     visualize.main(args)