We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7938293 commit 83436eeCopy full SHA for 83436ee
pytorch_lightning/callbacks/prediction_writer.py
@@ -54,7 +54,7 @@ class CustomWriter(BasePredictionWriter):
54
55
def __init__(self, output_dir: str, write_interval: str):
56
super().__init__(write_interval)
57
- self.output_dir
+ self.output_dir = output_dir
58
59
def write_on_batch_end(
60
self, trainer, pl_module: 'LightningModule', prediction: Any, batch_indices: List[int], batch: Any,
0 commit comments