Pretrained model doesn't need to inherit from nn.Module
MAYAS3 opened this issue · comments
since we us this class as a wrapper and we do not aim to optimize the pretrained models, we do not have to inherit from nn.module here
Uhm, no since we need to put the model in eval mode.
It seems not necessary, #27 ?
See #29
Oh I see... Right!