我是使用 StyleGAN 的新手。我刚刚尝试使用笔记本并收到“断言错误”。我不太明白如何解决它。
File "/usr/local/lib/python3.7/dist-packages/click/core.py", line 610, in invoke
return callback(*args, **kwargs)
File "/usr/local/lib/python3.7/dist-packages/click/decorators.py", line 21, in
new_func
return f(get_current_context(), *args, **kwargs)
File "train.py", line 575, in main
subprocess_fn(rank=0, args=args, temp_dir=temp_dir)
File "train.py", line 422, in subprocess_fn
training_loop.training_loop(rank=rank, **args)
File "/content/drive/MyDrive/colab-sg2-ada-pytorch/stylegan2-ada-
pytorch/training/training_loop.py", line 226, in training_loop
grid_size, images, labels = setup_snapshot_image_grid(training_set=training_set)
File "/content/drive/MyDrive/colab-sg2-ada-pytorch/stylegan2-ada-
pytorch/training/training_loop.py", line 63, in setup_snapshot_image_grid
images, labels = zip(*[training_set[i] for i in grid_indices])
File "/content/drive/MyDrive/colab-sg2-ada-pytorch/stylegan2-ada-
pytorch/training/training_loop.py", line 63, in <listcomp>
images, labels = zip(*[training_set[i] for i in grid_indices])
File "/content/drive/MyDrive/colab-sg2-ada-pytorch/stylegan2-ada-
pytorch/training/dataset.py", line 96, in __getitem__
assert list(image.shape) == self.image_shape
AssertionError