We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
zeroshampoo/distributed_shampoo.py
Line 55 in 65f24ff
If you want to compile, this line will have to be removed. torch compile cannot handle Python ints or floats that change.
Instead, def step(self): will become def step(self, calc_precondition: bool):
def step(self):
def step(self, calc_precondition: bool):
The text was updated successfully, but these errors were encountered:
No branches or pull requests
zeroshampoo/distributed_shampoo.py
Line 55 in 65f24ff
If you want to compile, this line will have to be removed. torch compile cannot handle Python ints or floats that change.
Instead,
def step(self):will becomedef step(self, calc_precondition: bool):The text was updated successfully, but these errors were encountered: