Skip to content

Commit

Permalink
fix readme
Browse files Browse the repository at this point in the history
  • Loading branch information
sebffischer committed Aug 22, 2024
1 parent 6023b3e commit a7b8b23
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion README.Rmd
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,7 @@ layer = list(
```

Next, we create a neural network that takes as input a `lazy_tensor` (`po("torch_ingress_num")`).
Next, we create a neural network that takes as input a `lazy_tensor` (`po("torch_ingress_ltnsr")`).
It first applies a linear layer and then repeats the above layer using the special `PipeOpTorchBlock`, followed by the network's head.
After that, we configure the loss, optimizer and the training parameters.
Note that `po("nn_linear_0")` is equivalent to `po("nn_linear", id = "nn_linear_0")` and we need this here to avoid ID clashes with the linear layer from `po("nn_block")`.
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -161,7 +161,7 @@ layer = list(
```

Next, we create a neural network that takes as input a `lazy_tensor`
(`po("torch_ingress_num")`). It first applies a linear layer and then
(`po("torch_ingress_ltnsr")`). It first applies a linear layer and then
repeats the above layer using the special `PipeOpTorchBlock`, followed
by the network’s head. After that, we configure the loss, optimizer and
the training parameters. Note that `po("nn_linear_0")` is equivalent to
Expand Down

0 comments on commit a7b8b23

Please sign in to comment.