(self, batch)
| 335 | return x |
| 336 | |
| 337 | def shared_step(self, batch): |
| 338 | x = self.get_input(batch, self.first_stage_key) |
| 339 | loss, loss_dict = self(x) |
| 340 | return loss, loss_dict |
| 341 | |
| 342 | def training_step(self, batch, batch_idx): |
| 343 | loss, loss_dict = self.shared_step(batch) |
no test coverage detected