mirror of
https://github.com/IBM/ai-privacy-toolkit.git
synced 2026-06-08 15:05:13 +02:00
Formatting
Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
parent
a46c4cad9e
commit
eeab625c3e
1 changed files with 0 additions and 1 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue