jdenholm/TorchTools

Add ``get_features`` method to CNN classification/regression models.

Closed this issue · 0 comments

jdenholm commented

Add a method for extracting only the features from the inputs.