link to torch installation instructions in readme

mlr-org · May 15, 2024 · 1c42f96 · 1c42f96
1 parent a8f9162
commit 1c42f96
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/README.Rmd b/README.Rmd
@@ -36,6 +36,8 @@ Deep Learning with torch and mlr3.
 
 ## Installation
 
+For more details on how to install {torch}, see [torch installation instruction](https://torch.mlverse.org/docs/articles/installation).
+
 ```{r eval = FALSE}
 # Install the development version from GitHub:
 pak::pak("mlr-org/mlr3torch")

diff --git a/README.md b/README.md
@@ -20,6 +20,9 @@ status](https://www.r-pkg.org/badges/version/mlr3torch)](https://CRAN.R-project.
 
 ## Installation
 
+For more details on how to install {torch}, see [torch installation
+instruction](https://torch.mlverse.org/docs/articles/installation).
+
 ``` r
 # Install the development version from GitHub:
 pak::pak("mlr-org/mlr3torch")
@@ -101,7 +104,7 @@ graph_mlp
 #> Graph with 8 PipeOps:
 #>                   ID         State            sccssors         prdcssors
 #>               <char>        <char>              <char>            <char>
-#>    torch_ingress_num <<UNTRAINED>>           nn_linear
+#>    torch_ingress_num <<UNTRAINED>>           nn_linear                  
 #>            nn_linear <<UNTRAINED>>             nn_relu torch_ingress_num
 #>              nn_relu <<UNTRAINED>>             nn_head         nn_linear
 #>              nn_head <<UNTRAINED>>          torch_loss           nn_relu
@@ -161,7 +164,7 @@ materialize(
   rbind = TRUE
 )
 #> torch_tensor
-#> (1,1,.,.) =
+#> (1,1,.,.) = 
 #>     0.0000    0.0000    0.0000    0.0000    0.0000
 #>     0.0000  200.9199  228.2500    8.2000    0.0000
 #>     0.0000    0.0000  196.7500    0.0000    0.0000