KeyError: 'rounds'

[NKUShaw]

Restart 0
MLP(
(_main): Sequential(
(0): Linear(in_features=392, out_features=256, bias=True)
(1): ReLU(inplace=True)
(2): Linear(in_features=256, out_features=256, bias=True)
(3): ReLU(inplace=True)
)
) TopMLP(
(lin1): Linear(in_features=256, out_features=1, bias=True)
(_main): Sequential(
(0): Linear(in_features=256, out_features=1, bias=True)
)
)
params received by train_func: {}
Traceback (most recent call last):
File "run_exp.py", line 225, in
main(flags)
File "run_exp.py", line 111, in main
res = train_func(*params)
File "FeAT/ColoredMNIST/train.py", line 1095, in ifeat_train
rounds = hparams['rounds']
KeyError: 'rounds'