Skip to content

The training question about diffste. #12

Open

Description

@thebestYezhang

Thanks for your great work! I have a question. When I train the diffste in my own dataset, the training Progress bar only prints out the loss of half of total training steps, and the loss of the rest half of total training steps is not show by the Progress bar. I carefully reviewed the training process, but couldn't find the problem. May I ask if you were in the same situation at training time?
image

Activity

  1. Question406 commented on Jul 16, 2023

    @Question406
    Collaborator

    Hi, actually, the training loss printed in the terminal is the loss for a single batch instead of the average loss over the entire dataset. For the details of what is printed, please see this link.

    The reason this progress bar is split in the middle of one epoch is the output during training, the line do_classifier_free_guidance; I guess this is some log information you print when you log some generated images.

  2. shaun-sheep111 commented on Dec 26, 2023

    @shaun-sheep111

    i want to say format of data,i want to train text editing

  3. shaun-sheep111 commented on Dec 26, 2023

    @shaun-sheep111

    when i training model,the path of log makes me very sad

  4. HHanTheKing commented on Jun 27, 2025

    @HHanTheKing

    Thanks for your great work! I have a question. When I train the diffste in my own dataset, the training Progress bar only prints out the loss of half of total training steps, and the loss of the rest half of total training steps is not show by the Progress bar. I carefully reviewed the training process, but couldn't find the problem. May I ask if you were in the same situation at training time? image

    Hello, I also want to use my own dataset for fine-tuning. What is the structure of your dataset (the training data provided by the blogger is quite complex). Is it in the form of a triple of mask-original image-image to be generated? How many images are there in each group? I hope to get your reply, thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions