Disabled for pipeline parallel training. See ``train_batch()``.
(self, *args, **kwargs)
| 1301 | return self._grad_layer_buf[idx].flatten()[:numel].view(shape) |
| 1302 | |
| 1303 | def forward(self, *args, **kwargs): |
| 1304 | """Disabled for pipeline parallel training. See ``train_batch()``. """ |
| 1305 | raise PipelineError("Only train_batch() is accessible in pipeline mode.") |
| 1306 | |
| 1307 | def backward(self, *args, **kwargs): |
| 1308 | """Disabled for pipeline parallel training. See ``train_batch()``. """ |
no test coverage detected