MCPcopy
hub / github.com/hpcaitech/ColossalAI / train.py

File train.py

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

Source from the content-addressed store, hash-verified

1import argparse
2
3import torch
4from data.bert_helper import SequenceParallelDataIterator, get_batch_for_sequence_parallel

Callers

nothing calls this directly

Calls 1

mainFunction · 0.70

Tested by

no test coverage detected