Skip to content

Inference time is longer than training time #301

Open
@Qusijia

Description

@Qusijia

My training time is much shorter than my reasoning time. Would you like to inquire if this situation is normal? Or what advice can you give me.

Training Code:
loss = priorTrainer.diffusion_prior(text_embed=text_embed, image_embed=image_embed, text_encodings=None)
priorTrainer.optimizer.zero_grad()
loss.backward()
priorTrainer.optimizer.step()
priorTrainer.scheduler.step()

Inference code:
prior.sample_add( text_embed = text_embed)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions