DenseRelationalCaptioning
DenseRelationalCaptioning copied to clipboard
Published
20 hours ago •
Dong-JinKim
Dong-JinKim
total train time
hello! thanks for your code~~Could you tell me how long you've been training the whole network?And how many GPU you use to run it? I think that the network is too large that maybe I cannot run it. I haven't run it yet.Looking forward to your reply!thanks!
Hello. Thank you for your interest in our code.
When training, the GPU consumption was about 12GB, and it took about 3~4 days for training.
I think you would need a GPU larger than 12GB for training a model. We used Nvidia TITAN X to train our model.
If your GPU is not large enough to run the code, I would like to suggest to reduce the size of the FC layers from 512 to 256 or 128.
Best regards,
