File size: 143 Bytes
b4efe93 | 1 2 3 4 5 6 7 8 9 | from dataclasses import dataclass
@dataclass
class AdamWOptimizerConfig:
_target_: str = "torch.optim.adamw.AdamW"
lr: float = 1e-6
|
b4efe93 | 1 2 3 4 5 6 7 8 9 | from dataclasses import dataclass
@dataclass
class AdamWOptimizerConfig:
_target_: str = "torch.optim.adamw.AdamW"
lr: float = 1e-6
|