You could use forward hooks as described e.g. here or manipulate the forward method directly by creating a new module deriving from the EfficientNet_b0 implementation.
1 Like
You could use forward hooks as described e.g. here or manipulate the forward method directly by creating a new module deriving from the EfficientNet_b0 implementation.