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
787 stars 38 forks source link

unrecognized arguments: --task --system ae_ddpm #6

Closed ansj11 closed 4 months ago

ansj11 commented 4 months ago

I am running python train_p_diff.py --task cifar100 --system ae_ddpm, but get following error

usage: train_p_diff.py [--help] [--hydra-help] [--version] [--cfg {job,hydra,all}] [--resolve] [--package PACKAGE] [--run] [--multirun] [--shell-completion] [--config-path CONFIG_PATH] [--config-name CONFIG_NAME] [--config-dir CONFIG_DIR] [--experimental-rerun EXPERIMENTAL_RERUN] [--info [{all,config,defaults,defaults-tree,plugins,searchpath}]] [overrides [overrides ...]] train_p_diff.py: error: unrecognized arguments: --task --system ae_ddpm

1zeryu commented 4 months ago

We load config file using 'hydra' package. The override must using '=', so you can change your command to python train_p_diff.py task=cifar100 system=ae_ddpm, If you have any questions feel free to ask us!

ansj11 commented 4 months ago

Thank you very much, I can run the code successifully!