2021年1月16日星期六

Issues with stylegan2 and args [closed]

I've been trying to train my own stylegan2 model but I keep having the same issue. I've gotten every other function to work but training. Every time I run it I get the same error.

  File "run_training.py", line 192, in <module>      run(**vars(args))    File "run_training.py", line 187, in main      run(**vars(args))    File "run_training.py", line 120, in run      dnnlib.submit_run(**kwargs)    File "/home/stylegan2/dnnlib/submission/submit.py", line 343, in submit_run      return farm.submit(submit_config, host_run_dir)    File "/home/stylegan2/dnnlib/submission/internal/local.py", line 22, in submit      return run_wrapper(submit_config)    File "/home/stylegan2/dnnlib/submission/submit.py", line 280, in run_wrapper      run_func_obj(**submit_config.run_func_kwargs    File "/home/stylegan2/training/training_loop.py", line 141, in training_loop      training_set = dataset.load_dataset(data_dir=dnnlib.convert_path(data_dir), verbose=True, **dataset_args)    File "/home/stylegan2/training/dataset.py", line 192, in load_dataset      dataset = dnnlib.util.get_obj_by_name(class_name)(**kwargs)    File "/home/stylegan2/training/dataset.py", line 53, in __init__      assert os.path.isdir(self.tfrecord_dir)   AssertionError  

After looking at the code I'm thinking its not taking my directory to the dataset correctly. When I change self.tfrecord_dir on line 53 in dataset.py to the directory itself it moves on until it needs the directory again. all the erros are with the dataset, but when I put it in the code it works in the spots I've tried, but I'd rather get it working the right way. Any help is greatly appreciated.

https://stackoverflow.com/questions/65756210/issues-with-stylegan2-and-args January 17, 2021 at 08:40AM

没有评论:

发表评论