search

liuruijin17 / LSTR

This is an official repository of End-to-end Lane Shape Prediction with Transformers.
BSD 3-Clause "New" or "Revised" License
647 stars 130 forks source link

论文中的420fps的batch设置为多少？ #60

Open xiuyangleiasp opened 3 years ago

xiuyangleiasp commented 3 years ago

请教下，论文中的420fps是在什么GPU上测试的，以及batch设置为多少？ 我在1070上和titanX上，batch=16时的fps，分别为1000fps和1600fps。

liuruijin17 commented 3 years ago

Hi, thanks for your interest!

Emm, your computer is so good. Our batch16 only gets 420 on my PC. CPU: AMD Ryzen 1900x GPU: 2x1080Ti DIMM: 4x16 DDR4 3200MHz SSD: 2x2T SAMSUNG 970 EVO Plus 2T This is my ancient working PC:rofl:

plz583585760 commented 2 years ago

请问一下有尝试过复现吗，训练到后面loss值到多少呢，我的到1.8左右就不下降了，出来的模型检测效果也不太好，没找到原因。

xiuyangleiasp commented 2 years ago

@plz583585760 我这边之前按照默认配置，复现acc是没有问题的，loss没在意具体是多少；后来就换了backbone，做QAT了。

plz583585760 commented 2 years ago

@plz583585760 我这边之前按照默认配置，复现acc是没有问题的，loss没在意具体是多少；后来就换了backbone，做QAT了。

为什么我复现的时候chunksizes只能设置为1才能训练，我用的是windows系统，显卡是3060，训练完后准确率只有85左右，所以很困惑不知道哪里出现了问题，您的chunksizes设置为多少啊？

plz583585760 commented 2 years ago

@plz583585760 我这边之前按照默认配置，复现acc是没有问题的，loss没在意具体是多少；后来就换了backbone，做QAT了。

为什么我复现的时候chunksizes只能设置为1才能训练，我用的是windows系统，显卡是3060，训练完后准确率只有85左右，所以很困惑不知道哪里出现了问题，您的chunksizes设置为多少啊？

xiuyangleiasp commented 2 years ago

@plz583585760 我这边之前按照默认配置，复现acc是没有问题的，loss没在意具体是多少；后来就换了backbone，做QAT了。

为什么我复现的时候chunksizes只能设置为1才能训练，我用的是windows系统，显卡是3060，训练完后准确率只有85左右，所以很困惑不知道哪里出现了问题，您的chunksizes设置为多少啊？

chunksizes没改，设的是16；我一般都是Ubuntu训练，windows系统没办法把GPU占满，不过这个模型训练不需要那么多的显存就是了