ssd-tf2
ssd-tf2 copied to clipboard
修复 ssd.py 中导致内存溢出的问题
取消使用@tf.function装饰器 经测试 在ssd.py的get_pred函数中 该装饰器会引起内存溢出 注释即可 不会十分影响效率 可能与 issue#6 有关 train_eager.py中未测试 故不进行commit 可以注释进行测试
emm这个@tf.function会提高速度。你有比较过吗,差距很大
emm这个@tf.function会提高速度。你有比较过吗,差距很大
可是如果不采用此方法,无法完成训练,有别的解决方式吗?
取消使用@tf.function装饰器 经测试 在ssd.py的get_pred函数中 该装饰器会引起内存溢出 注释即可 不会十分影响效率 可能与 issue#6 有关 train_eager.py中未测试 故不进行commit 可以注释进行测试
我注释完运行,发现内存的占用仍然在不断的上升
如果直接使用train.py应该没用到tf.function(),我真的不太明白为啥会上升……因为我这里一点事都没