FeAT
FeAT copied to clipboard
KeyError: 'rounds'
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