Skip to content

Instantly share code, notes, and snippets.

@GitHubEmploy

GitHubEmploy/pytorch_2.py

Created December 30, 2021 20:56
Show Gist options
  • Save GitHubEmploy/0b2454bd4d25bb0e6d89a419dfe69047 to your computer and use it in GitHub Desktop.
Save GitHubEmploy/0b2454bd4d25bb0e6d89a419dfe69047 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pytorch_2.py
import torch
import torchvision
import torchvision.transforms as transforms
num_epochs = 10
batch_size = 32
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment