research/gam/gam/experiments/run_train_gam.py [84:103]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
flags.DEFINE_string(
    'model_agr', 'mlp',
    'Model type for the agreement model. Options are: `mlp`, `cnn`, '
    '`wide_resnet`.')
flags.DEFINE_float('learning_rate_cls', 0.001,
                   'Initial learning rate of the classification model.')
flags.DEFINE_float('learning_rate_agr', 0.001,
                   'Initial learning rate of the agreement model.')
flags.DEFINE_float('learning_rate_decay_cls', None,
                   'Learning rate decay factor for the classification model.')
flags.DEFINE_float('learning_rate_decay_agr', None,
                   'Learning rate decay factor for the agreement model.')
flags.DEFINE_float('lr_decay_rate_cls', None,
                   'Learning rate decay rate for the classification model.')
flags.DEFINE_integer('lr_decay_steps_cls', None,
                     'Learning rate decay steps for the classification model.')
flags.DEFINE_float('lr_decay_rate_agr', None,
                   'Learning rate decay rate for the agreement model.')
flags.DEFINE_integer('lr_decay_steps_agr', None,
                     'Learning rate decay steps for the agreement model.')
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



research/gam/gam/experiments/run_train_gam_graph.py [69:88]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
flags.DEFINE_string(
    'model_agr', 'mlp',
    'Model type for the agreement model. Options are: `mlp`, `cnn`, '
    '`wide_resnet`.')
flags.DEFINE_float('learning_rate_cls', 0.001,
                   'Initial learning rate of the classification model.')
flags.DEFINE_float('learning_rate_agr', 0.001,
                   'Initial learning rate of the agreement model.')
flags.DEFINE_float('learning_rate_decay_cls', None,
                   'Learning rate decay factor for the classification model.')
flags.DEFINE_float('learning_rate_decay_agr', None,
                   'Learning rate decay factor for the agreement model.')
flags.DEFINE_float('lr_decay_rate_cls', None,
                   'Learning rate decay rate for the classification model.')
flags.DEFINE_integer('lr_decay_steps_cls', None,
                     'Learning rate decay steps for the classification model.')
flags.DEFINE_float('lr_decay_rate_agr', None,
                   'Learning rate decay rate for the agreement model.')
flags.DEFINE_integer('lr_decay_steps_agr', None,
                     'Learning rate decay steps for the agreement model.')
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -