ssd-tf2 icon indicating copy to clipboard operation
ssd-tf2 copied to clipboard

修复 ssd.py 中导致内存溢出的问题

Open CurryEx opened this issue 4 years ago • 4 comments

取消使用@tf.function装饰器 经测试 在ssd.py的get_pred函数中 该装饰器会引起内存溢出 注释即可 不会十分影响效率 可能与 issue#6 有关 train_eager.py中未测试 故不进行commit 可以注释进行测试

CurryEx avatar Apr 13 '21 03:04 CurryEx

emm这个@tf.function会提高速度。你有比较过吗,差距很大

bubbliiiing avatar Apr 15 '21 08:04 bubbliiiing

emm这个@tf.function会提高速度。你有比较过吗,差距很大

可是如果不采用此方法,无法完成训练,有别的解决方式吗?

Lee-Akita avatar Apr 16 '21 02:04 Lee-Akita

取消使用@tf.function装饰器 经测试 在ssd.py的get_pred函数中 该装饰器会引起内存溢出 注释即可 不会十分影响效率 可能与 issue#6 有关 train_eager.py中未测试 故不进行commit 可以注释进行测试

我注释完运行,发现内存的占用仍然在不断的上升

Lee-Akita avatar Apr 16 '21 03:04 Lee-Akita

如果直接使用train.py应该没用到tf.function(),我真的不太明白为啥会上升……因为我这里一点事都没

bubbliiiing avatar Apr 21 '21 06:04 bubbliiiing