5cedbc4a7c09285b2df2d8678d0449d231a1feeb,catalyst/runners/runner.py,Runner,predict_loader,#,295

Before Change

        Yields:
            bathes with model predictions
        
        if isinstance(fp16, bool) and fp16:
            fp16 = {"opt_level": "O1"}

        if model is not None:
            self.model = model
        assert self.model is not None

After Change

        Yields:
            bathes with model predictions
        
        fp16 = _resolve_bool_fp16(fp16)

        if model is not None:
            self.model = model
        assert self.model is not None

Instances