Autoencoder with fastai V2

I put a small example on how to construct an autoencoder using V2.
This is a Image2Image model using the DataBlockAPI
It runs on colab, you can view it here
I struggled a little bit, so it may be useful for someone else.


Great work! Could you put it over here in the projects thread? :slight_smile:

Thanks very much for your input. I am using fastai to built my own AE and this has been very helpful!

I updated the notebooks to V2 release (was on beta V2).

1 Like

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()’
‘arch = create_body(xresnet18(pretrained=True), n_in=1).cuda()’


‘get_y = lambda o: o,’
‘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:

1 Like

Yeah, this is very old. It does not take into account that fastai now integrates timm, and other cool stuff!