Fix ordering of parameters.
parent
eb10075b16
commit
6277c2db59
|
|
@ -1303,8 +1303,8 @@
|
|||
},
|
||||
"adamw_weight_decay": {
|
||||
"title": "AdamW Weight Decay",
|
||||
"type": "float",
|
||||
"default": 1e-2
|
||||
"type": "number",
|
||||
"default": 0.01
|
||||
},
|
||||
"mixed_precision": {
|
||||
"title": "Mixed Precision",
|
||||
|
|
|
|||
|
|
@ -108,8 +108,8 @@ class DreamboothParameters(BaseModel):
|
|||
lr_scheduler: str = "constant"
|
||||
lr_warmup_steps: int = 500
|
||||
max_token_length: int = 75
|
||||
adamw_weight_decay: float = 1e-2
|
||||
mixed_precision: str = "no"
|
||||
adamw_weight_decay: float = 1e-2
|
||||
model_name: str = ""
|
||||
num_train_epochs: int = 100
|
||||
pad_tokens: bool = True
|
||||
|
|
|
|||
|
|
@ -541,8 +541,8 @@ def on_ui_tabs():
|
|||
db_lr_scheduler,
|
||||
db_lr_warmup_steps,
|
||||
db_max_token_length,
|
||||
db_adamw_weight_decay,
|
||||
db_mixed_precision,
|
||||
db_adamw_weight_decay,
|
||||
db_model_path,
|
||||
db_num_train_epochs,
|
||||
db_pad_tokens,
|
||||
|
|
@ -677,8 +677,8 @@ def on_ui_tabs():
|
|||
db_lr_scheduler,
|
||||
db_lr_warmup_steps,
|
||||
db_max_token_length,
|
||||
db_adamw_weight_decay,
|
||||
db_mixed_precision,
|
||||
db_adamw_weight_decay,
|
||||
db_num_train_epochs,
|
||||
db_pad_tokens,
|
||||
db_pretrained_vae_name_or_path,
|
||||
|
|
|
|||
Loading…
Reference in New Issue