Adding a new example to README.:

kaanaksit · Dec 1, 2023 · 74d7334 · 74d7334
1 parent d48e225
commit 74d7334
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -127,6 +127,8 @@ model_mlp = odak.learn.models.multi_layer_perceptron(
                                                      bias = False,
                                                      model_type = 'conventional'
                                                     )
+
+
 optimizer = torch.optim.AdamW(model_mlp.parameters(), lr = 1e-3)
 loss_function = torch.nn.MSELoss()
 for epoch in range(10000):
@@ -138,6 +140,7 @@ for epoch in range(10000):
     optimizer.step()
 print('Training loss: {}'.format(loss.item()))
 
+
 for item_id, item in enumerate(pos_x1):
     torch.no_grad()
     ground_truth = x1[item_id]