Jit scripting #204

Unanswered

zaccharieramzi asked this question in Q&A

zaccharieramzi
Jan 30, 2024

Hi !

Has anyone tried to jit script functions implemented using torchopt, and if so did you manage to make it work?

At the moment I am facing an issue similar to:

RuntimeError: 
undefined value torchopt:
    ) -> torch.Tensor:
        optimizer = torchopt.FuncOptimizer(torchopt.adam(lr=self.lr))
                    ~~~~~~~~ <--- HERE
    
        with torch.enable_grad():

Replies: 0 comments

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment