c16df44669828b8566c7bfb87d385c73cd74e460,catalyst/dl/runner/core.py,Runner,trace,#,296

Before Change

            output: Path = logdir / "trace"
            output.mkdir(exist_ok=True, parents=True)

            out_model = str(output / filename)

            torch.jit.save(result, out_model)

        return result

After Change

            )

        // Restore previous state of the model
        getattr(self.model, "train" if _is_training else "eval")()
        utils.set_requires_grad(self.model, _requires_grad)
        self.model.to(_device)

        return traced_model

Instances