search

baset-sarker / yolov7_in_kv260

1 stars 0 forks source link

some questions about the config and device of the project #1

Open SM2XWe1L opened 2 months ago

SM2XWe1L commented 2 months ago

Can i implement this project on Vitis-ai 2.5 cpu? And can you publish your final model's performance like fps? Thank you very much!

baset-sarker commented 2 months ago

I have tried to run on Vitis-ai 2.5-cpu. The model training works, but it takes time to train. I am working on the performance of the model, and I will publish it ASAP. Thank you

SM2XWe1L commented 2 months ago

Thanks for your kind answer ,but i encounter a problem when i want to run the QAT script with yolov7:

image

Have you ever met this problem? Thirsty for your reply.

baset-sarker commented 2 months ago

I did not encounter it, but it is a problem with the CUDA and PyTorch versions. you can have a look here : https://github.com/pytorch/pytorch/issues/30664

SM2XWe1L commented 2 months ago

I did not encounter it, but it is a problem with the CUDA and PyTorch versions. you can have a look here : pytorch/pytorch#30664

OK, thanks! I will try with it.

baset-sarker commented 2 months ago

May I know what version of Ubuntu you are using? I have found many different issues with Ubuntu 22.04. switching to Ubuntu 20.04 solved most of those issues.