dzhang97 comments

Repositories
Issues
Comments

Results 1 comments of


                                            dzhang97

Memory-Usage of GPU increases when training.

the issue lies in the variables saved for backward in the psroipooling layer. Between pytorch 0.3 and 0.4, the functionality was changed to require using the save_for_backward() function instead of...