search

lliuz / ARFlow

The official PyTorch implementation of the paper "Learning by Analogy: Reliable Supervision from Transformations for Unsupervised Optical Flow Estimation".
MIT License
251 stars 50 forks source link

Information on GPU and Memory #25

Closed st164137 closed 3 years ago

st164137 commented 3 years ago

Hi,

Could you please let me know which GPU have you used to train the model. When I run the program with some modifications cuda is going out of memory.

lliuz commented 3 years ago

I trained it with GTX1080ti / TITAN X On 12/22/2020 00:23, st164137 wrote: Hi, Could you please let me know which GPU have you used to train the model. When I run the program with some modifications cuda is going out of memory.

—You are receiving this because you are subscribed to this thread.Reply to this email directly, view it on GitHub, or unsubscribe. [ { "@context": "http://schema.org", "@type": "EmailMessage", "potentialAction": { "@type": "ViewAction", "target": "https://github.com/lliuz/ARFlow/issues/25", "url": "https://github.com/lliuz/ARFlow/issues/25", "name": "View Issue" }, "description": "View this Issue on GitHub", "publisher": { "@type": "Organization", "name": "GitHub", "url": "https://github.com" } } ]

st164137 commented 3 years ago

Could you please also let me know the training time ?