torch_cg
torch_cg copied to clipboard
Reached max iter bug
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!