Screenshot 2021-05-15 at 2.19.39 AM1440×900 216 KB

Here is my model:
<bound method Module.state_dict of model(
(hidden_layers): ModuleList(
(0): Linear(in_features=3, out_features=10, bias=True)
(1): Linear(in_features=10, out_features=10, bias=True)
)
(output_layers): ModuleList(
(0): Linear(in_features=10, out_features=4, bias=True)
(1): Linear(in_features=10, out_features=4, bias=True)
)
)>