ml3/experiments/run_sine_regression_exp.py [10:29]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
if __name__ == "__main__":
    exp_cfg = {}
    exp_cfg['seed'] = 0
    exp_cfg['num_train_tasks'] = 1
    exp_cfg['num_test_tasks'] = 10
    exp_cfg['n_outer_iter'] = 500
    exp_cfg['n_gradient_steps_at_test'] = 100
    exp_cfg['inner_lr'] = 0.001
    exp_cfg['outer_lr'] = 0.001

    exp_cfg['model'] = {}
    exp_cfg['model']['in_dim'] = 1
    exp_cfg['model']['hidden_dim'] = [100, 10]

    exp_cfg['metaloss'] = {}
    exp_cfg['metaloss']['in_dim'] = 2
    exp_cfg['metaloss']['hidden_dim'] = [50, 50]

    model_arch_str = str(exp_cfg['model']['hidden_dim'])
    meta_arch_str = "{}".format(exp_cfg['metaloss']['hidden_dim'])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



ml3/sine_regression_task.py [182:201]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
if __name__ == "__main__":
    exp_cfg = {}
    exp_cfg['seed'] = 0
    exp_cfg['num_train_tasks'] = 1
    exp_cfg['num_test_tasks'] = 10
    exp_cfg['n_outer_iter'] = 500
    exp_cfg['n_gradient_steps_at_test'] = 100
    exp_cfg['inner_lr'] = 0.001
    exp_cfg['outer_lr'] = 0.001

    exp_cfg['model'] = {}
    exp_cfg['model']['in_dim'] = 1
    exp_cfg['model']['hidden_dim'] = [100, 10]

    exp_cfg['metaloss'] = {}
    exp_cfg['metaloss']['in_dim'] = 2
    exp_cfg['metaloss']['hidden_dim'] = [50, 50]

    model_arch_str = str(exp_cfg['model']['hidden_dim'])
    meta_arch_str = "{}".format(exp_cfg['metaloss']['hidden_dim'])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -