search

gsig / actor-observer

ActorObserverNet code in PyTorch from "Actor and Observer: Joint Modeling of First and Third-Person Videos", CVPR 2018
GNU General Public License v3.0
76 stars 9 forks source link

Out of memory. #2

Open happsky opened 6 years ago

happsky commented 6 years ago

Hi, thank you for sharing this amazing code. How many GPUs are you used to train the model, and what type of the GPUs? Thank you in advance!

gsig commented 6 years ago

I used an individual Titan X gpu for training/inference. Let me know if you have any questions.

happsky commented 6 years ago

Thanks for your reply. I have another question that what's the difference between baseline_resnet152charades.py and baseline_resnet152imagenet.py?

gsig commented 6 years ago

baseline_resnet152charades.py uses a resnet152 trained on the charades dataset

baseline_resnet152imagenet.py uses a resnet152 trained on imagenet

Hope that helps!

On Fri, Jun 29, 2018, 10:42 PM Happsky notifications@github.com wrote:

Thanks for your reply. I have another question that what's the difference between baseline_resnet152charades.py and baseline_resnet152imagenet.py?

— You are receiving this because you commented. Reply to this email directly, view it on GitHub https://github.com/gsig/actor-observer/issues/2#issuecomment-401511974, or mute the thread https://github.com/notifications/unsubscribe-auth/ADbmNToXzFy5D2ls6CJ3bXzQEaON79P5ks5uBuWagaJpZM4U03sW .