update galore args
This commit is contained in:
parent
33a4c24a8a
commit
0ac6b40a47
|
@ -178,7 +178,7 @@ class GaloreArguments:
|
||||||
default=0.25,
|
default=0.25,
|
||||||
metadata={"help": "GaLore scale."},
|
metadata={"help": "GaLore scale."},
|
||||||
)
|
)
|
||||||
galore_proj_type: Literal["std"] = field(
|
galore_proj_type: Literal["std", "reverse_std", "right", "left", "full"] = field(
|
||||||
default="std",
|
default="std",
|
||||||
metadata={"help": "Type of GaLore projection."},
|
metadata={"help": "Type of GaLore projection."},
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue