Created a toy model to fit into Learner for image classification, but getting 0 accuracy. Has anyone faced a similar problem before?
model = nn.Sequential(
nn.Linear(24, 36),nn.ReLU(),
nn.Linear(36, 36),nn.ReLU(),
nn.Linear(36, 10),nn.ReLU(),
Lambda(lambda x: x.view(x.size(0),-1))
)
learn = Learner(data, model, metrics=accuracy)
Thanks in advance, Neo