import

[jwijffels]

torch needs to be loaded or some functions from torch are not imported yet

library(torchvision)
m <- model_resnet18()
Error in nn_conv2d(3, self$inplanes, kernel_size = 7, stride = 2, padding = 3,  : 
  could not find function "nn_conv2d"
  
library(torch)
m <- model_resnet18()
m
An `nn_module` containing 11,689,512 parameters.

-- Modules -----------------------------------------------------------------------------------------------------------------------------
* conv1: <nn_conv2d> #9,408 parameters
* bn1: <nn_batch_norm2d> #128 parameters
* relu: <nn_relu> #0 parameters
* maxpool: <nn_max_pool2d> #0 parameters
* layer1: <nn_sequential> #147,968 parameters
* layer2: <nn_sequential> #525,568 parameters
* layer3: <nn_sequential> #2,099,712 parameters
* layer4: <nn_sequential> #8,393,728 parameters
* avgpool: <nn_adaptive_avg_pool2d> #0 parameters
* fc: <nn_linear> #513,000 parameters