The problem in my case was "train_steps" in the model json file. Default is 0. The notebook sets it to 401000 which works.
def __init__(self, channels, stride=1):
super(ConvBlock, self).__init__()
self.convolution = nn.Conv2d(channels//stride, channels, 3, stride=stride, padding=1)
self.normalization = nn.BatchNorm2d(channels)
#self.activation = nn.LeakyReLU()
def forward(self, x):
x =
# x = self.convolution(x)
# x = self.normalization(x)
# x = self.activation(x)
x = self.convolution(x)
x = self.normalization(x)
x = self.activation(x)
return x
def __init__(self, stride=1):
super(Resblock, self).__init__()
self.res = nn.Sequential(
nn.Conv2d(9, 9, 3, stride=stride, padding=1),
nn.BatchNorm2d(9),
nn.LeakyReLU(0.2)
)
self.output = nn.MaxPool2d(2)
def forward(self, x):
return self.res(x)