Skip to content

Instantly share code, notes, and snippets.

@J3698
Created March 22, 2021 15:51
Show Gist options
  • Save J3698/c60174391f6ce6c134355581d6719b81 to your computer and use it in GitHub Desktop.
Save J3698/c60174391f6ce6c134355581d6719b81 to your computer and use it in GitHub Desktop.
class VGG19Encoder(nn.Module):
def __init__(self):
super().__init__()
self.feats = nn.Sequential(*self.extract_vgg19_pretrained_layers())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment