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

hi, i only have a titan for 12G,can i run the train.py? #6

Open
dzz416 opened this issue Aug 2, 2021 · 8 comments
Open

hi, i only have a titan for 12G,can i run the train.py? #6

dzz416 opened this issue Aug 2, 2021 · 8 comments

Comments

@dzz416
Copy link

dzz416 commented Aug 2, 2021

No description provided.

@dzz416 dzz416 changed the title 作者你好,数据集是已经编码好的呀,只有解码文件,能提供一下编码文件吗,谢谢啦大佬 hi, i only have a titan for 12G,can i run the train.py? Aug 2, 2021
@hjSim
Copy link
Collaborator

hjSim commented Aug 2, 2021

Yes, you can. We also have tried to train the network with NVIDIA Titan Xp and the memory usage was about 7GB for patch size 384x384 and 11GB for patch size 512x512. All other experiment settings are the same as the description in our paper.

@dzz416
Copy link
Author

dzz416 commented Aug 3, 2021

Yes, you can. We also have tried to train the network with NVIDIA Titan Xp and the memory usage was about 7GB for patch size 384x384 and 11GB for patch size 512x512. All other experiment settings are the same as the description in our paper.

I have a another problem, when i download the encode_train.tar ,it always failed in the end, can i download it in another way ?

@hjSim
Copy link
Collaborator

hjSim commented Aug 3, 2021

Since the uncompressed original training dataset (as .png files) is about 240GB, we could not upload the uncompressed one online.
What is the error when you run 'mp4_decoding.py' we provided?

@dzz416
Copy link
Author

dzz416 commented Aug 3, 2021

The training set in the link is 14g, and every time I download it to 13.9g, it is banned.

@JihyongOh
Copy link
Owner

@dzz416 ,
We have successfully re-downloaded it through the same link just now.
Can you check your network condition, please?

@dzz416
Copy link
Author

dzz416 commented Aug 8, 2021

I have downloaded the train data successfully,thank you ! I 'm just getting start , Can you tell me the exact format of the training input data and output data of this network model and the function of decode.py?tanks again

@dzz416
Copy link
Author

dzz416 commented Aug 9, 2021

And how long does it take to train?

@hjSim
Copy link
Collaborator

hjSim commented Aug 10, 2021

@dzz416
We use '.png' image file for both input and output data.
'.png' files are extracted from the encoded files we provided, via dropbox link.
'mp4_decoding.py' file can be found in the same dropbox link as well.
Training takes about a half-day with an NVIDIA TITAN RTX as described in our paper. It depends on the experiment system.
Most of information can be found in our paper or github readme page
Thank you!

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

3 participants