File convnext_example.py

torch/distributed/_tensor/examples/convnext_example.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	"""
2	The following example demonstrates how to train a ConvNeXt model
3	with intermediate activations sharded across mutliple GPUs via DTensor
4	"""

nothing calls this directly

spawnMethod · 0.80

no test coverage detected