Closed jwijffels closed 3 years ago
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
torch needs to be loaded or some functions from torch are not imported yet