all_methods_on_mnist.py 文件源码

def _check_adam():
    for _mode in HO_MODES[:2]:
        for _model in IMPLEMENTED_MODEL_TYPES[1:2]:
            _model_kwargs = {'dims': [None, 300, 300, None]}
            tf.reset_default_graph()

            # set random seeds!!!!
            np.random.seed(1)
            tf.set_random_seed(1)

            experiment('test_with_model_' + _model,
                       collect_data=False, hyper_iterations=3, mode=_mode, epochs=3,
                       optimizer=rf.AdamOptimizer,
                       optimizer_kwargs={'lr': tf.Variable(.001, name='eta_adam')},
                       model=_model,
                       model_kwargs=_model_kwargs,
                       set_T=100,
                       )