Title could be vague but here is the problem:
I have a codes = net.encoder(x)
line which performs a single pass on encoder with its forward function defined as
def forward(self, x):
x = self.input_layer(x)
and self.input_layer
is defined as
self.input_layer = Sequential(Conv2d(3, 64, (3, 3), 1, 1, bias=False),
BatchNorm2d(64),
PReLU(64))
apparently on my linux machine with torch 2.0.1 this runs fine but in colab, it doesn’t throws error but can’t pass beyond x = self.input_layer(x)
which i checked with a print statement afterwards. How to debug this?