connecting_the_dots icon indicating copy to clipboard operation
connecting_the_dots copied to clipboard

Published 20 hours ago • verified publisher icon autonomousvision

census

Open meteorshowers opened this issue 5 years ago • 1 comments

Does the pytorch census loss the same with the cuda census loss? Why not use the pytorch census loss? Thanks a lot! @yiyiliao @barnjamin

meteorshowers avatar Dec 14 '19 03:12 meteorshowers

I was wondering this too. Turns out that the PyTorch implementation requires a lot of memory. A Titan RTX with 24GB memory was not enough for a batch size of 2.

mehrab2603 avatar Aug 13 '20 19:08 mehrab2603