i run the test i got this issue
⊱ ──────ஓ๑♡๑ஓ ────── ⊰ 🎵 hhey, arguments are here if you need to check 🎵 checkpoint_pattern: ./checkpoints/pattern.pth checkpoint_color: ./checkpoints/color.pth device: cuda prn: True color_only: True pattern_only: False input: ./imgs/non-makeup.png style: ./imgs/style-1.png alpha: 0.5 savedir:
raceback (most recent call last):
File "main.py", line 27, in
model = Makeup(args)
File "D:\AI\MakeUp\makeup.py", line 19, in init
self.pattern.test_model(args.checkpoint_pattern)
File "D:\AI\MakeUp\utils\models.py", line 46, in test_model
torch.load(path),
File "C:\Users\This PC\miniconda3\envs\makeup\lib\site-packages\torch\serialization.py", line 795, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "C:\Users\This PC\miniconda3\envs\makeup\lib\site-packages\torch\serialization.py", line 1012, in _legacy_load
result = unpickler.load()
File "C:\Users\This PC\miniconda3\envs\makeup\lib\site-packages\torch\serialization.py", line 828, in find_class
return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'segmentation_models_pytorch.unet'
i run the test i got this issue
⊱ ──────ஓ๑♡๑ஓ ────── ⊰ 🎵 hhey, arguments are here if you need to check 🎵 checkpoint_pattern: ./checkpoints/pattern.pth checkpoint_color: ./checkpoints/color.pth device: cuda prn: True color_only: True pattern_only: False input: ./imgs/non-makeup.png style: ./imgs/style-1.png alpha: 0.5 savedir: