Bug report while training

[eecoder-dyf]

When running train_unicon.py, it reports the following error:

  File "/mnt/ssd1/dyf/local/UniCon/utils/utils.py", line 192, in set_unicon_config_train
    patch.set_unicon_config(unet, "attn_config", attn_config)
AttributeError: module 'patch.patch' has no attribute 'set_unicon_config'. Did you mean: 'update_unicon_config'?

I have checked the repo, there does not exist set_unicon_config in file patch/patch.py.

[lixirui142]

Thanks for your bug report. It has been fixed.