example pytorch scripts load data to gpu

Thomas McFadden answered on September 10, 2023 Popularity 1/10 Helpfulness 1/10

answer example pytorch scripts load data to gpu

example pytorch scripts load data to gpu

Comment

import torch
from my_classes import Dataset


# CUDA for PyTorch
use_cuda = torch.cuda.is_available()
device = torch.device("cuda:0" if use_cuda else "cpu")
torch.backends.cudnn.benchmark = True

# Parameters
params = {'batch_size': 64,
          'shuffle': True,
          'num_workers': 6}
max_epochs = 100

# Datasets
partition = # IDs
labels = # Labels

# Generators
training_set = Dataset(partition['train'], labels)
training_generator = torch.utils.data.DataLoader(training_set, **params)

validation_set = Dataset(partition['validation'], labels)
validation_generator = torch.utils.data.DataLoader(validation_set, **params)

# Loop over epochs
for epoch in range(max_epochs):
    # Training
    for local_batch, local_labels in training_generator:
        # Transfer to GPU
        local_batch, local_labels = local_batch.to(device), local_labels.to(device)

        # Model computations
        [...]

    # Validation
    with torch.set_grad_enabled(False):
        for local_batch, local_labels in validation_generator:
            # Transfer to GPU
            local_batch, local_labels = local_batch.to(device), local_labels.to(device)

            # Model computations
            [...]

Popularity 1/10 Helpfulness 1/10 Language typescript

Source: stanford.edu

Tags: gpu load pytorch typescript

Link to this answer
Share Copy Link

Contributed on Sep 10 2023

Thomas McFadden

0 Answers Avg Quality 2/10

example pytorch scripts load data to gpu

Contents

More Related Answers

example pytorch scripts load data to gpu

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.