NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
https://arxiv.org/abs/2007.03898
Other
999 stars 163 forks source link

i am getting this error when try to run train.py file #19

Open haideralimughal opened 3 years ago

haideralimughal commented 3 years ago

_norm_anneal=False, weight_decay_norm_init=10.0) 01/04 09:33:02 AM (Elapsed: 00:00:23) param size = 6.482514M 01/04 09:33:02 AM (Elapsed: 00:00:23) groups per scale: [10], total_groups: 10 01/04 09:33:02 AM (Elapsed: 00:00:23) epoch 0 /usr/bin/nvidia-modprobe: unrecognized option: "-s"

ERROR: Invalid commandline, please run /usr/bin/nvidia-modprobe --help for usage information.

/usr/bin/nvidia-modprobe: unrecognized option: "-s"

ERROR: Invalid commandline, please run /usr/bin/nvidia-modprobe --help for usage information.