cascade.core.PredictionWriter

class cascade.core.PredictionWriter(output_dir)[source]

Bases: BasePredictionWriter

Custom prediction writer to enable multi-device prediction