Question

我正在尝试在Google Colab上运行StyleGAN。如果遵循在Linux上运行代码所使用的相同步骤，则会收到以下错误。您对我应该更改的内容有什么建议吗？

Creating the run dir: results/00002-sgan-ffhq-8gpu
Copying files to the run dir
dnnlib: Running training.training_loop.training_loop() on localhost...
Streaming data using training.dataset.TFRecordDataset...
Traceback (most recent call last):
  File "train.py", line 190, in <module>
    main()
  File "train.py", line 185, in main
    dnnlib.submit_run(**kwargs)
  File "/content/stylegan/dnnlib/submission/submit.py", line 290, in submit_run
    run_wrapper(submit_config)
  File "/content/stylegan/dnnlib/submission/submit.py", line 242, in run_wrapper
    util.call_func_by_name(func_name=submit_config.run_func_name, submit_config=submit_config, **submit_config.run_func_kwargs)
  File "/content/stylegan/dnnlib/util.py", line 257, in call_func_by_name
    return func_obj(*args, **kwargs)
  File "/content/stylegan/training/training_loop.py", line 146, in training_loop
    training_set = dataset.load_dataset(data_dir=config.data_dir, verbose=True, **dataset_args)
  File "/content/stylegan/training/dataset.py", line 234, in load_dataset
    dataset = dnnlib.util.get_obj_by_name(class_name)(**adjusted_kwargs)
  File "/content/stylegan/training/dataset.py", line 70, in __init__
    assert os.path.isdir(self.tfrecord_dir)
AssertionError

以下是笔记本的链接：https://gist.github.com/albusdemens/11c609d7db26d8006fed186ec102471b

AssertionError在Colab上运行StyleGAN

0 个答案: