ComfyUI Node
LLM Arguments
A ComfyUI node in 大模型学校(llm_schools)/模型训练(Model Training) with 10 inputs and 1 output.
LLM Arguments
- args
◄output_dirresults►
◄eval_strategyepoch►
◄learning_rate0.0000►
◄per_device_train_batch_size1►
◄per_device_eval_batch_size1►
◄num_train_epochs3►
◄weight_decay0.01►
◄save_steps1000►
◄save_total_limit2►
◄is_enabletrue►
Category大模型学校(llm_schools)/模型训练(Model Training)
Inputs (10)
| Name | Type | Default | Description |
|---|---|---|---|
| output_dir | STRING | results | — |
| eval_strategy | STRING | epoch | — |
| learning_rate | FLOAT | 0.00000–1 | — |
| per_device_train_batch_size | INT | 1 | — |
| per_device_eval_batch_size | INT | 1 | — |
| num_train_epochs | INT | 3 | — |
| weight_decay | FLOAT | 0.010–1 | — |
| save_steps | INT | 1000 | — |
| save_total_limit | INT | 2 | — |
| is_enable | BOOLEAN | true | — |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| args | ARGS | — |