Thank you @sgugger. This fixed the problem. Now I have another one:
learn.fit_one_cycle(1, 3e-2, moms=(0.8,0.7))
It gives an error at the validation stage - the default metrics accuracy is not appropriate may be:
~/anaconda2/envs/fastai.1/lib/python3.6/site-packages/fastai/metrics.py in accuracy(input, targs)
37 input = input.argmax(dim=-1).view(n,-1)
38 targs = targs.view(n,-1)
---> 39 return (input==targs).float().mean()
40
41 def error_rate(input:Tensor, targs:Tensor)->Rank0Tensor:
RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'other'
I try to fix this with:
learn.metrics = accuracy_thresh
But I get an error:
TypeError: 'function' object is not iterable
Edit: a list of metrics is expected. This works:
learn.metrics = [accuracy_thresh]