Unlike a model defined with nn.Sequential
, TabularModel
doesn’t allow indexing to select the layers.
How do I save & load only the body weights (but not the head weights) of a TabularModel
?
Use case: transfer learning. (The features are exactly the same in both models. Only the targets are different.)
Has anyone managed to do transfer learning with tabular data? Any hints/pointers would be greatly appreciated.