Spandrel gives your project support for various PyTorch architectures meant for AI Super-Resolution, restoration, and inpainting. Based on the model support implemented in chaiNNer.
MIT License
103
stars
7
forks
source link
Suggest a valid size requirement if size requirement test fails #211
The assert_size_requirements test already collects data on which image sizes are valid and invalid for a given model. While it previously only displayed the collected data, it will now try to guess a size requirement that fits the data. This should make it easier to add support for new archs, because assert_size_requirements will just tell you the size requirements of a model now.
The
assert_size_requirements
test already collects data on which image sizes are valid and invalid for a given model. While it previously only displayed the collected data, it will now try to guess a size requirement that fits the data. This should make it easier to add support for new archs, becauseassert_size_requirements
will just tell you the size requirements of a model now.