"torchvision/csrc/ops/cuda/nms_kernel.cu" did not exist on "0564df4348812604ea1d21976c81676fdaf6f9be"
-
Deepak Narayanan authored
Also includes following changes for inter-layer model-parallel implementation: - Refactoring of model implementations - Training loop changes to support inter-layer communication using `ring_exchange` - New groups for inter-layer communication - Checkpoint changes - Command line arguments
7abd3e90
