Unverified Commit 8ea50d99 authored by Super Daniel's avatar Super Daniel Committed by GitHub
Browse files

[hotfix] pass a parameter. (#2288)

* [autockpt] make it work.

* [autockpt] linearize / merge shape-consistency nodes.

* [autockpt] considering parameter and optimizer weights.

* [hotfix] pass a parameter.
parent 22e947f9
...@@ -50,7 +50,7 @@ class CheckpointSolverRotor(CheckpointSolverBase): ...@@ -50,7 +50,7 @@ class CheckpointSolverRotor(CheckpointSolverBase):
optim_multiplier (float, optional): The multiplier of extra weight storage for the optim_multiplier (float, optional): The multiplier of extra weight storage for the
``torch.optim.Optimizer``. Default to 1.0. ``torch.optim.Optimizer``. Default to 1.0.
""" """
super().__init__(graph, free_memory, True, cnode) super().__init__(graph, free_memory, True, cnode, optim_multiplier)
self.memory_slots = memory_slots self.memory_slots = memory_slots
# construct chain # construct chain
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment