modelscope / modelscope/DiffSynth-Studio
训练异常
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 13.1k
- Forks
- 1.3k
- Avg merge
- 13h 12m
- Merged PRs (30d)
- 45
Description
我用一张A800 80G的显卡训练 加载了一天 还是一直没开始训练 想问一下这个是什么原因呢,lora的是elign
CUDA_VISIBLE_DEVICES=1 accelerate launch --num_processes=1 examples/qwen_image/model_training/train.py
--dataset_base_path
--dataset_metadata_path
--data_file_keys "image,eligen_entity_masks"
--max_pixels 1048576
--dataset_repeat 20
--model_id_with_origin_paths "Qwen/Qwen-Image:transformer/diffusion_pytorch_model*.safetensors,Qwen/Qwen-Image:text_encoder/model*.safetensors,Qwen/Qwen-Image:vae/diffusion_pytorch_model.safetensors"
--learning_rate 1e-4
--num_epochs 5
--remove_prefix_in_ckpt "pipe.dit."
--output_path "./models/train/Qwen-Image-EliGen_lora"
--lora_base_model "dit"
--lora_target_modules "to_q,to_k,to_v,add_q_proj,add_k_proj,add_v_proj,to_out.0,to_add_out,img_mlp.net.2,img_mod.1,txt_mlp.net.2,txt_mod.1"
--lora_rank 32
--extra_inputs "eligen_entity_masks,eligen_entity_prompts"
--use_gradient_checkpointing
--find_unused_parameters
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with examples/qwen_image/model_training/train.py and run the supplied accelerate command, checking its startup output and the attached image for where initialization stops. The issue needs a reproducible diagnosis of why training has not begun, including the relevant logs or configuration details and a clear confirmation of the cause.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Quiet
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100