Open jlevy44 opened 4 years ago
I had made an attempt to incorporate nvidia-apex https://github.com/NVIDIA/apex into this repository and airlab https://github.com/jlevy44/airlab to no avail. For now, just raising a NotImplementedError when selecting the half option.
@bitebitebit
Still debugging, but seems incorporate different image levels quite well, which was not done in previous models. Still some memory issues as well once you get to larger images in the pyramid.
Will add more details later.