MCPcopy Create free account
hub / github.com/mlco2/codecarbon / pytorch-multigpu-example.py

File pytorch-multigpu-example.py

examples/pytorch-multigpu-example.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import numpy as np
2import torch
3from torch import nn, optim
4from torch.utils.data import DataLoader

Callers

nothing calls this directly

Calls 5

EmissionsTrackerClass · 0.90
CNNClass · 0.85
flushMethod · 0.80
startMethod · 0.45
stopMethod · 0.45

Tested by

no test coverage detected