Skip to content

Instantly share code, notes, and snippets.

@thunderInfy

thunderInfy/simclr-linear-classifier.py

Created April 4, 2020 08:59
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save thunderInfy/9b83c642de97b7efab751f0b0825aafb to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save thunderInfy/9b83c642de97b7efab751f0b0825aafb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
simclr-linear-classifier.py
class LinearNet(nn.Module):
def __init__(self):
super(LinearNet, self).__init__()
self.fc1 = torch.nn.Linear(50, 5)
def forward(self, x):
x = self.fc1(x)
return(x)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment