ngruver / decon-hnn Goto Github PK

NN
{'C': 5,
 'bs': 200,
 'data_seed': 0,
 'device': 'cuda:0',
 'loss': 'l2',
 'lr': 0.005,
 'n_systems': 1000,
 'net_cfg': {'hidden_size': 128},
 'net_seed': 0,
 'num_epochs': 2000,
 'regen': False,
 'tau': 10.0,
 'weight_decay': 0.0001}
Traceback (most recent call last):
  File "/home/ubuntu/decon-hnn/toy_systems.py", line 68, in <module>
    Fire(main)
  File "/home/ubuntu/anaconda3/envs/check/lib/python3.9/site-packages/fire/core.py", line 141, in Fire
    component_trace = _Fire(component, args, parsed_flag_args, context, name)
  File "/home/ubuntu/anaconda3/envs/check/lib/python3.9/site-packages/fire/core.py", line 466, in _Fire
    component, remaining_args = _CallAndUpdateTrace(
  File "/home/ubuntu/anaconda3/envs/check/lib/python3.9/site-packages/fire/core.py", line 681, in _CallAndUpdateTrace
    component = fn(*varargs, **kwargs)
  File "/home/ubuntu/decon-hnn/toy_systems.py", line 60, in main
    trainer = make_trainer(**cfg, network=net, body=body,
TypeError: make_trainer() got an unexpected keyword argument 'loss'