s440058/ium_440058
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add scared fileobserver
All checks were successful
s440058-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
piotr6789 2021-05-24 15:22:46 +02:00
parent ff040678cc
commit 1841e9ea1a
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
scared-example-file.py
View File

@ -60,7 +60,6 @@ def script(num_epochs, batch_size, _run):
output_dim = 1
model = LogisticRegressionModel(input_dim, output_dim)
model.load_state_dict(torch.load('diabetes.pth'))
criterion = torch.nn.BCELoss(reduction='mean')
optimizer = torch.optim.SGD(model.parameters(), lr = learning_rate)