support layerwise galore
Former-commit-id: d43a4da0947897d0be3f62fad3107754d4c89f2b
This commit is contained in:
@@ -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