def resnet18(gen):
model = ResNetS(BasicBlock, [2, 2, 2, 2],
avg_pool_kernel=(7, 1),
n_channels=gen.n_channels,
num_classes=gen.n_classes)
return model
def resnet18(gen):
model = ResNetS(BasicBlock, [2, 2, 2, 2],
avg_pool_kernel=(7, 1),
n_channels=gen.n_channels,
num_classes=gen.n_classes)
return model