support layerwise galore
This commit is contained in:
1 parent
18ffce36b5
commit
8664262cde
14 files changed
+109
-51
No files matched your search
@@ -182,6 +182,10 @@ class GaloreArguments:
|
||||
default="std",
|
||||
metadata={"help": "Type of GaLore projection."},
|
||||
)
|
||||
galore_layerwise: bool = field(
|
||||
default=False,
|
||||
metadata={"help": "Whether or not to enable layer-wise update to further save memory."},
|
||||
)
|
||||
|
||||
|
||||
@dataclass
|
||||
|
||||
Reference in new issue
Block a user