Open hamidriasat opened 2 years ago
Hi thanks for the PR, but it seems like that the new model you added might not work with other functionality of the repo. For e.g. it is not added in the all_models.py , hence cli tools wont be able to use it.
Hi @divamgupta, thanks for your reply. New changes have been made in code to make it compatible with rest of the repo functionality. Please test it with cli tools. I hope it works this time, if anything is missing please let me know.
Following changes have been made in the current commit:
all_models.py
unet3_plus.py
to make sure local methods are not accessible to outside classes/objectstest_models.py
to make sure it's easily readable test_models
method of test_models.py
file
Two changes