Reuse Resnet Backbone from U-net

Deep Learning
1

The situation is the following. A U-Net with a resnet34 backbone is trained. Is there a way to reuse the fine-tuned backbone part of the U-Net to use it to perform more transfer-learning for other tasks? If so, how do I get this “sub-network” from the U-Net?

2

You can do it. You can simply choose a part of your entire architecture. The code would look like this.

list(learn.model.modules())[0]
4 Likes
3

Thanks a lot for the answer @VishnuSubramanian, and sorry for my late response!

1 Like