Can I get information about the training time?

CVIR / CoMix

This repository contains the official implementation of CoMix (NeurIPS 2021) https://arxiv.org/pdf/2110.15128.pdf.

Apache License 2.0

21 stars 8 forks source link

Can I get information about the training time? #3

Closed Backdrop9019 closed 2 years ago

Backdrop9019 commented 2 years ago

Thank you for the interesting research. Currently, I'm using 4 RTX A5000s to run the code, but it's very slow to take more than 5 minutes per liter. Is this normal? Can I get information about the training time? If so, I would be very grateful

Backdrop9019 commented 2 years ago

I don't know why, but when I installed using conda env.yml, model.cuda() was very slow. When I updated the torch, it worked out well. :)