Thanks tcapelle for sharing your autoencoder solution. Interesting solution.
Maybe there has been version updates. When running it now and installing newest versions I had to make following two modifications:
‘arch = create_body(xresnet18, n_in=1).cuda()’
to
‘arch = create_body(xresnet18(pretrained=True), n_in=1).cuda()’
and
‘get_y = lambda o: o,’
to
‘get_y = named_function_to_make_learn_export_work,’
Without named function I ended up to error when saving the learner. This seems to be generic fastai issue and not just with my environment: