search

NUS-HPC-AI-Lab / Neural-Network-Parameter-Diffusion

We introduce a novel approach for parameter generation, named neural network parameter diffusion (p-diff), which employs a standard latent diffusion model to synthesize a new set of parameters
828 stars 42 forks source link

No module named 'core.system.explore' #3

Closed ansj11 closed 7 months ago

ansj11 commented 7 months ago

I am running the demo code bash cifar100_resnet18_k200.sh, but i get ModuleNotFoundError: No module named 'core.system.explore'. Is there missing Neural-Network-Diffusion/core/system/explore.py?

1zeryu commented 7 months ago

The explore.py file is the one we used during the exploration of p-diff, and it has already been deleted here. The bug in code core/system/init.py `from .base import from .ddpm import from .encoder import * from .vae import VAESystem from .ae_ddpm import AE_DDPM from .explore import Explore

systems = { 'encoder': EncoderSystem, 'ddpm': DDPM, 'vae': VAESystem, 'ae_ddpm': AE_DDPM, 'explore': Explore, }`

should change to (just delete explore)

` from .base import from .ddpm import from .encoder import * from .vae import VAESystem from .ae_ddpm import AE_DDPM

systems = { 'encoder': EncoderSystem, 'ddpm': DDPM, 'vae': VAESystem, 'ae_ddpm': AE_DDPM, }`

Thanks for the feedback.