Disabled for pipeline parallel training. See ``train_batch()``.
(self, *args, **kwargs)
| 1317 | return self._grad_layer_buf[idx].flatten()[:numel].view(shape) |
| 1318 | |
| 1319 | def forward(self, *args, **kwargs): |
| 1320 | """Disabled for pipeline parallel training. See ``train_batch()``. """ |
| 1321 | raise PipelineError("Only train_batch() is accessible in pipeline mode.") |
| 1322 | |
| 1323 | def backward(self, *args, **kwargs): |
| 1324 | """Disabled for pipeline parallel training. See ``train_batch()``. """ |
no test coverage detected