About wavelet_model.py

[hongchu098]

Thanks for your open source codes. I tried to run the train_wavelet.py and got the error message "ModuleNotFoundError: No module named 'models.wavelet_model'", which not existed at 'models' folder. A similar model existed at wavelet_cycle.py, but it seems to lack the FAM loss.

Could you please update the latest model code file like wavelet_model.py?

[zhaohaoyu376]

Sorry. I add this file to Github. You can try again.