Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/parameters.py

Created August 31, 2019 14:14
Show Gist options
  • Save omarsar/e7a7606bb8054093e8e559788c4a2ed7 to your computer and use it in GitHub Desktop.
Save omarsar/e7a7606bb8054093e8e559788c4a2ed7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
parameters.py
learning_rate = 0.001
num_epochs = 5
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model = MyModel()
model = model.to(device)
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment