Closed zarmondo11 closed 2 years ago
Hi please revise the following part:
model = build_model(task="Deblurring")
because the default model if for Dehazing:
def build_model(task = "Dehazing"):
model_mod = importlib.import_module(f'maxim.models.{_MODEL_FILENAME}')
model_configs = ml_collections.ConfigDict(_MODEL_CONFIGS)
model_configs.variant = _MODEL_VARIANT_DICT[task]
model = model_mod.Model(**model_configs)
return model
thanks a lot. <3
Hello. I try to run this rep on google colab and it works fine with Enhancement pre-trained model but when I want to load Deblurring model and use predict() to get output, this error appears...
what's the problem? is it with pre-trained models? if yes, how can I fix it or make my own model?