torch_cg icon indicating copy to clipboard operation
torch_cg copied to clipboard

Reached max iter bug

Open fabian-sp opened this issue 3 years ago • 0 comments

Hi, when running your code, I spotted this small bug: https://github.com/sbarratt/torch_cg/blob/7504429f13cc66d6a4b0d5767f561abb557a2a1e/torch_cg/cg_batch.py#L104

Here, it should print "optimal" in the case when optimal, and else "reached max iter". So just switching if and else.

Thanks and best!

fabian-sp avatar Jan 25 '22 21:01 fabian-sp