Open
Description
In cnn_archs notebook you mentioned we need to monkey patch the forward pass(excluding final layers). But it seems like you've also excluded initial transform_input part as well.
if self.transform_input:
x = x.clone()
x[:, 0] = x[:, 0] * (0.229 / 0.5) + (0.485 - 0.5) / 0.5
x[:, 1] = x[:, 1] * (0.224 / 0.5) + (0.456 - 0.5) / 0.5
x[:, 2] = x[:, 2] * (0.225 / 0.5) + (0.406 - 0.5) / 0.5
Although I tried to add these lines to your notebook and it ran without any issue. But is there any specific reason to skip this part ?
Metadata
Metadata
Assignees
Labels
No labels