The train function uses standard print statements for logging, but the main block of Fabric examples (03_fabric-vit.py and later) uses fabric.print. This inconsistency makes the output harder to follow, particularly in multi-GPU scenarios where only the main process should print certain information. All logging should utilize fabric.print when Fabric is used.
The
trainfunction uses standardprintstatements for logging, but the main block of Fabric examples (03_fabric-vit.py and later) usesfabric.print. This inconsistency makes the output harder to follow, particularly in multi-GPU scenarios where only the main process should print certain information. All logging should utilizefabric.printwhen Fabric is used.