search

fanyix / STMN

[ECCV 2018] Video Object Detection with an Aligned Spatial-Temporal Memory
http://fanyix.cs.ucdavis.edu/project/stmn/project.html
Other
118 stars 16 forks source link

out of memery #6

Closed zhaowujie closed 5 years ago

zhaowujie commented 5 years ago

I only have a 12GB gpu, how to modify the parameters to avoid the problem of out of memory @fanyix

shady-cs15 commented 5 years ago

+1 It trains for a few hours and then it runs out of memory. Is there some memory leak happening? Screenshot 2019-04-05 at 5 51 13 PM

fanyix commented 5 years ago

Hi, you can try changing "timestep_per_batch" in training config (config.lua) to a smaller value to reduce memory consumption.

zhaowujie commented 5 years ago

I'll try , thanks a lot!