How long does the training take and where is the .pt file

[handsomecong001]

This is a great code for the diffusion model. I trained for over 10,000 steps with Colab. Will the training stop automatically? And I can't find the .pt file. And I can't find the /tmp/ file either.

[Zanghu-ze]

Hi @handsomecong001 The first question the code will stop automatically. In the default parameters, it seems like the code set the total steps.
As for the second question, the default path which restores the pt file is the tmp. You can check the whole machine and find where the tmp file is. It is usually located at the /tmp. The outermost folder, not inside the improved-diffusion/ folder. hoping these things will help you!

[adistomar]

You need to set the OPENAI_LOGDIR environment variable to the path where you want the .pt file to go:

export OPENAI_LOGDIR=/path/to/dir