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.

Sign up for GitHub

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

Jump to bottom

Training mode error #30

Open
drahmad89 opened this issue Feb 3, 2020 · 4 comments
Open

Training mode error #30

drahmad89 opened this issue Feb 3, 2020 · 4 comments

Comments

@drahmad89
Copy link

I have added a training module and I noticed when forwarding the hourglass88 with torch.no_grad it works just fine but when removing torch.no_grad the gpu (12GB) run out of memory just by forwarding one image sample. I get this message:

RuntimeError: CUDA out of memory. Tried to allocate 46.00 MiB (GPU 0; 10.91 GiB total capacity; 10.21 GiB already allocated; 44.75 MiB free; 92.41 MiB cached).

It seems by the paper that you training not 1 but 4 images per mini-batch.
@drahmad89 drahmad89 changed the title Training mode erro Training mode error Feb 6, 2020
@ashishgupta2598
Copy link

Hii, Is the training code available!!!
Thanks

@chenzeng11
Copy link

Hi, could you share your training code? Thank you!

@HHeracles
Copy link

@drahmad89 Hi, could you share your training code? Thank you!

1 similar comment
@XFGforever
Copy link

@drahmad89 Hi, could you share your training code? Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
5 participants
@chenzeng11 @drahmad89 @ashishgupta2598 @HHeracles @XFGforever