search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

2080ti上面训练，显存溢出 #21

Closed deep-practice closed 4 years ago

deep-practice commented 4 years ago

2080ti上面，网络输入大小416，把batch size设为8都会显存溢出，这是正常的吗。 image image

deep-practice commented 4 years ago

Solved

liuxundd commented 2 years ago

我也遇到这个问题 ，请问怎么解决的？