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
没有评论:
发表评论