Formatting

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-09-21 12:48:47 +03:00
parent a46c4cad9e
commit eeab625c3e

View file

@ -979,7 +979,6 @@ def test_minimizer_pytorch(data_three_features):
out = self.fc4(out)
return self.classifier(out)
base_est = pytorch_model(2, 3)
criterion = nn.CrossEntropyLoss()
optimizer = optim.Adam(base_est.parameters(), lr=0.01)