ColossalAI/examples/tutorial/large_batch_optimizer/requirements.txt

3 lines
36 B
Plaintext

colossalai >= 0.1.12
torch >= 1.8.1