MCPcopy Index your code
hub / github.com/hpcaitech/ColossalAI / train.py

File train.py

examples/tutorial/large_batch_optimizer/train.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2import torch.nn as nn
3from torchvision.models import resnet18
4from tqdm import tqdm

Callers

nothing calls this directly

Calls 1

mainFunction · 0.70

Tested by

no test coverage detected