Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

training iterations #1

Open
tgc1997 opened this issue Apr 18, 2023 · 11 comments
Open

training iterations #1

tgc1997 opened this issue Apr 18, 2023 · 11 comments

Comments

@tgc1997
Copy link

tgc1997 commented Apr 18, 2023

Nice work, I have a question about the training.
In your paper, you train the model for 5 * 10^5 iterations. However, N_iters in config.txt is set to 5000001. What is the correct number of parameters?

@circle-DING
Copy link

I have the same question.I modify parameters in configs/lego/test1_events.txt,changed N_iters (5000001 to 500001) and lrate_ Decay_ Steps (50000000 to 5000000), however, the image rendered based on readme is pure pink purple.Have you resolved this question?

@r00tman
Copy link
Owner

r00tman commented Oct 23, 2023

Hello, sorry for the delay.
The correct value is 5 * 10^5 iterations as in the paper. The value in the config (5*10^6) is irrelevant as we stopped training manually at 5 * 10^5 iterations.

@r00tman
Copy link
Owner

r00tman commented Oct 23, 2023

@circle-DING , could you please share the results?

@r00tman
Copy link
Owner

r00tman commented Oct 23, 2023

I'll share pretrained models today for easier reproduction

@circle-DING
Copy link

@circle-DING , could you please share the results?

Due to limited computing resources, I'm sorry that I didn't train for 510^6 iterations. I stopped training at slightly over 510^5 iterations, which happens to be the correct approach you suggested.

@circle-DING
Copy link

I'll share pretrained models today for easier reproduction

Thank you for your reply and help!

@520jz
Copy link

520jz commented Mar 1, 2024

Hello, sorry for the delay. The correct value is 5 * 10^5 iterations as in the paper. The value in the config (5*10^6) is irrelevant as we stopped training manually at 5 * 10^5 iterations.

However, I found that if the number of iterations is set at 5x10^5 of the paper, the PSNR will not achieve the effect of the paper. I trained the lego scene with four 3090s, iterated 1 million times, and the PSNR was just 20, but in the paper it was 25

@520jz
Copy link

520jz commented Mar 6, 2024

However, I found that if the number of iterations is set at 5x10^5 of the paper, the PSNR will not achieve the effect of the paper. I trained the lego scene with four 3090s, iterated 1 million times, and the PSNR was just 20, but in the paper it was 25

However, I found that if the number of iterations is set at 5x10^5 of the paper, the PSNR will not achieve the effect of the paper. I trained the lego scene with four 3090s, iterated 1 million times, and the PSNR was just 20, but in the paper it was 25

@520jz
Copy link

520jz commented Mar 6, 2024

@circle-DING , could you please share the results?

However, I found that if the number of iterations is set at 5x10^5 of the paper, the PSNR will not achieve the effect of the paper. I trained the lego scene with four 3090s, iterated 1 million times, and the PSNR was just 20, but in the paper it was 25

@circle-DING
Copy link

circle-DING commented Mar 6, 2024 via email

@r00tman
Copy link
Owner

r00tman commented Mar 6, 2024

Hello, thank you for your question and sorry for the delay.

Could you briefly write how you compute PSNR on your side?

Just to be safe, let me briefly recap the evaluation procedure from my side.

The correct number of iterations is 5 * 10^5.

The logged PSNR should NOT be used for evaluation as it is derived from the training event-based loss of the current mini-batch.

The numbers reported in the paper are computed after training for about 5 * 10^5 iterations using the procedure described in metric/README.md.

To help the process, please also find the converged pretrained models uploaded here (models.zip).

I hope this helps. If you have more questions, I'm available and will respond rather quickly.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants