gaozhihan / gaozhihan/PreDiff

Training Time of Prediff

Open
#12 2 comments 0 reactions 1 assignee Claimed by @gaozhihan View on GitHub
good first issue question
Dominant language
Python
Stars
159
Forks
10
PR merge metrics
No merged PRs in 30d

Description

Could you provide the actual training time cost of the three models? According to the appendix, VAE, LDM, and KCN need 200, 1000, and 200 training epochs, respectively.

I tried on 3090 Ti and the experiment time costs are as follows.

4***3090** | Epochs | 1gpu | 2gpu | 4gpu | 4**gpu** Estimation (Day)
-- | -- | -- | -- | -- | --
VAE-338MB | 200 | 2h | 1h 52min | 1h20min | 11.11
LDM-548MB | 1000 | 2h | 1h16min | 53min | 36.81
KCN-35MB | 200 | 30min | 15min | 8min | 1.11

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.