aryadomain
/

all_code_base

Model card Files Files and versions

all_code_base / lrm /flux /trainer /optimizers /adamw.py

aryadomain's picture

Add files using upload-large-folder tool

b4efe93 verified 4 months ago

History Blame Contribute Delete

143 Bytes

	from dataclasses import dataclass


	@dataclass
	class AdamWOptimizerConfig:
	_target_: str = "torch.optim.adamw.AdamW"
	lr: float = 1e-6