I am trying to access weights of a neural network model built by torch on databricks.
the code :
import torch import torch.nn as nn import numpy