StyleGAN выдает ошибку утверждения после запуска train.py - PullRequest
Новая
       4

StyleGAN выдает ошибку утверждения после запуска train.py

0 голосов
/ 12 июля 2020

Вот ошибка, которую я все время получаю 

WARNING:tensorflow:From C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\tflib\tfutil.py:34: The name tf.Dimension is deprecated. Please use tf.compat.v1.Dimension instead.

WARNING:tensorflow:From C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\tflib\tfutil.py:74: The name tf.variable_scope is deprecated. Please use tf.compat.v1.variable_scope instead.

WARNING:tensorflow:From C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\tflib\tfutil.py:128: The name tf.Session is deprecated. Please use tf.compat.v1.Session instead.

Creating the run dir: results\00002-sgan-ffhq-1gpu
Copying files to the run dir
dnnlib: Running training.training_loop.training_loop() on localhost...
WARNING:tensorflow:From C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\tflib\tfutil.py:97: The name tf.get_default_session is deprecated. Please use tf.compat.v1.get_default_session instead.

WARNING:tensorflow:From C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\tflib\tfutil.py:109: The name tf.set_random_seed is deprecated. Please use tf.compat.v1.set_random_seed instead.

WARNING:tensorflow:From C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\tflib\tfutil.py:132: The name tf.ConfigProto is deprecated. Please use tf.compat.v1.ConfigProto instead.

Streaming data using training.dataset.TFRecordDataset...
Traceback (most recent call last):
  File "train.py", line 191, in <module>
    main()
  File "train.py", line 186, in main
    dnnlib.submit_run(**kwargs)
  File "C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\submission\submit.py", line 290, in submit_run
    run_wrapper(submit_config)
  File "C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\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 "C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\dnnlib\util.py", line 257, in call_func_by_name
    return func_obj(*args, **kwargs)
  File "C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\training\training_loop.py", line 146, in training_loop
    training_set = dataset.load_dataset(data_dir=config.data_dir, verbose=True, **dataset_args)
  File "C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\training\dataset.py", line 234, in load_dataset
    dataset = dnnlib.util.get_obj_by_name(class_name)(**adjusted_kwargs)
  File "C:\Users\716058\Desktop\Python stuff\Machine Learning\stylegan-master\training\dataset.py", line 70, in __init__
    assert os.path.isdir(self.tfrecord_dir)
AssertionError

Я очень новичок в styleGAN, но я проверил, и каталог в моем train.py правильный, любая помощь будет потрясающей, спасибо!

...