Получение ошибок при тренировке с тензорным потоком для адаптивной передачи стиля - PullRequest
0 голосов
/ 19 февраля 2019

Я далеко от моей глубины здесь.Я пытаюсь обучить модель передачи адаптивного стиля с моими собственными изображениями и набором данных, связанных в статье github: https://github.com/CompVis/adaptive-style-transfer.Через 4 часа компьютер звучит так, как будто все еще работает, но окно терминала не изменилось.Пожалуйста, посмотрите код ниже - кто-нибудь может мне помочь исправить ошибки загрузки?Спасибо.

CUDA_VISIBLE_DEVICES=1 python main.py \
>                  --model_name=model_lukea2 \
>                  --batch_size=1 \
>                  --phase=train \
>                  --image_size=768 \
>                  --lr=0.0002 \
>                  --dsr=0.8 \
>                  --ptcd=./data_large \
>                  --ptad=./data/lukea


2019-02-19 08:39:15.681527: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
WARNING:tensorflow:From /anaconda3/lib/python3.7/site-packages/tensorflow/python/framework/op_def_library.py:263: colocate_with (from tensorflow.python.framework.ops) is deprecated and will be removed in a future version.
Instructions for updating:
Colocations handled automatically by placer.
WARNING:tensorflow:From /anaconda3/lib/python3.7/site-packages/tensorflow/python/ops/math_ops.py:3066: to_int32 (from tensorflow.python.ops.math_ops) is deprecated and will be removed in a future version.
Instructions for updating:
Use tf.cast instead.
  0%|                                                                                                                                                                                                                                           | 0/132 [00:00<?, ?it/s]a/abbey 0
Category a/abbey can't be found in path ./data_large/a/abbey. Skip it.
a/arch 1
                                                                                                                                                                                                                                                                       a/amphitheater 2█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 462273.95it/s]
                                                                                                                                                                                                                                                                       a/aqueduct 3█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 475555.45it/s]
  3%|██████▉                                                                                                                                                                                                                            | 4/132 [00:00<00:04, 30.20it/s]a/arena/rodeo 4█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 494996.58it/s]
                                                                                                                                                                                                                                                                       a/athletic_field/outdoor 5███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 463223.55it/s]
                                                                                                                                                                                                                                                                       b/badlands 6█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 496038.60it/s]
  5%|████████████                                                                                                                                                                                                                       | 7/132 [00:00<00:05, 24.87it/s]b/balcony/exterior 7████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 499905.13it/s]
                                                                                                                                                                                                                                                                       b/bamboo_forest 8████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 481329.36it/s]
  7%|███████████████▍                                                                                                                                                                                                                   | 9/132 [00:00<00:06, 20.11it/s]b/barn 9████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 481086.44it/s]
                                                                                                                                                                                                                                                                       b/barndoor 10████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 499405.14it/s]
                                                                                                                                                                                                                                                                       b/baseball_field 11██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 489897.22it/s]
  9%|████████████████████▌                                                                                                                                                                                                             | 12/132 [00:00<00:05, 22.17it/s]b/basilica 12███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 470720.06it/s]
Category b/basilica can't be found in path ./data_large/b/basilica. Skip it.
b/bayou 13
Category b/bayou can't be found in path ./data_large/b/bayou. Skip it.
b/beach 14
                                                                                                                                                                                                                                                                       b/beach_house 15█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 504693.28it/s]
                                                                                                                                                                                                                                                                       b/beer_garden 16█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 488709.92it/s]
 13%|█████████████████████████████                                                                                                                                                                                                     | 17/132 [00:00<00:04, 26.15it/s]b/boardwalk 17██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 482714.24it/s]
                                                                                                                                                                                                                                                                       b/boathouse 18███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 487335.77it/s]
                                                                                                                                                                                                                                                                       b/botanical_garden 19████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 459599.39it/s]
 15%|██████████████████████████████████▏                                                                                                                                                                                               | 20/132 [00:00<00:04, 26.59it/s]b/bullring 20███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 499250.58it/s]

Я должен удалить кусок кода из-за ограничения количества символов здесь.Пропустить до конца:

                                                                                                                                                                                                                                                                       a/apartment-building/outdoor 128█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 439903.51it/s]
Category a/apartment-building/outdoor can't be found in path ./data_large/a/apartment-building/outdoor. Skip it.
a/artists_loft 129
 98%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▌   | 130/132 [00:04<00:00, 34.47it/s]b/building_facade 130███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 417177.64it/s]
                                                                                                                                                                                                                                                                       c/cemetery 131███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 467467.34it/s]
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 132/132 [00:04<00:00, 26.57it/s]
Finished. Constructed Places2 dataset of 624077 images.█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5000/5000 [00:00<00:00, 450003.65it/s]
Art dataset contains 13 images.
Processes are started.
Process Process-2:
Traceback (most recent call last):
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 297, in _bootstrap
    self.run()
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 99, in run
    self._target(*self._args, **self._kwargs)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 75, in initialize_batch_worker
    batch = self.get_batch(augmentor=augmentor, batch_size=batch_size)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 50, in get_batch
    image = scipy.misc.imread(name=random.choice(self.dataset), mode='RGB')
  File "/anaconda3/lib/python3.7/site-packages/numpy/lib/utils.py", line 101, in newfunc
    return func(*args, **kwds)
  File "/anaconda3/lib/python3.7/site-packages/scipy/misc/pilutil.py", line 164, in imread
    im = Image.open(name)
  File "/anaconda3/lib/python3.7/site-packages/PIL/Image.py", line 2657, in open
    % (filename if filename else fp))
OSError: cannot identify image file './data/lukea/.DS_Store'
Start training.
 [*] Reading latest checkpoint from folder ./models/model_lukea2/checkpoint.
 [*] Reading latest checkpoint from folder ./models/model_lukea2/checkpoint_long.
 [!] Load failed...
  0%|                                                                                                                                                                                                                           | 7/300000 [02:29<1669:29:21, 20.03s/it]Process Process-6:
Traceback (most recent call last):
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 297, in _bootstrap
    self.run()
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 99, in run
    self._target(*self._args, **self._kwargs)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 75, in initialize_batch_worker
    batch = self.get_batch(augmentor=augmentor, batch_size=batch_size)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 50, in get_batch
    image = scipy.misc.imread(name=random.choice(self.dataset), mode='RGB')
  File "/anaconda3/lib/python3.7/site-packages/numpy/lib/utils.py", line 101, in newfunc
    return func(*args, **kwds)
  File "/anaconda3/lib/python3.7/site-packages/scipy/misc/pilutil.py", line 164, in imread
    im = Image.open(name)
  File "/anaconda3/lib/python3.7/site-packages/PIL/Image.py", line 2657, in open
    % (filename if filename else fp))
OSError: cannot identify image file './data/lukea/.DS_Store'
  0%|                                                                                                                                                                                                                          | 15/300000 [05:19<1639:12:54, 19.67s/it]Process Process-8:
Traceback (most recent call last):
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 297, in _bootstrap
    self.run()
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 99, in run
    self._target(*self._args, **self._kwargs)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 75, in initialize_batch_worker
    batch = self.get_batch(augmentor=augmentor, batch_size=batch_size)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 50, in get_batch
    image = scipy.misc.imread(name=random.choice(self.dataset), mode='RGB')
  File "/anaconda3/lib/python3.7/site-packages/numpy/lib/utils.py", line 101, in newfunc
    return func(*args, **kwds)
  File "/anaconda3/lib/python3.7/site-packages/scipy/misc/pilutil.py", line 164, in imread
    im = Image.open(name)
  File "/anaconda3/lib/python3.7/site-packages/PIL/Image.py", line 2657, in open
    % (filename if filename else fp))
OSError: cannot identify image file './data/lukea/.DS_Store'
  0%|                                                                                                                                                                                                                          | 27/300000 [09:01<1531:32:34, 18.38s/it]Process Process-4:
Traceback (most recent call last):
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 297, in _bootstrap
    self.run()
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 99, in run
    self._target(*self._args, **self._kwargs)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 75, in initialize_batch_worker
    batch = self.get_batch(augmentor=augmentor, batch_size=batch_size)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 50, in get_batch
    image = scipy.misc.imread(name=random.choice(self.dataset), mode='RGB')
  File "/anaconda3/lib/python3.7/site-packages/numpy/lib/utils.py", line 101, in newfunc
    return func(*args, **kwds)
  File "/anaconda3/lib/python3.7/site-packages/scipy/misc/pilutil.py", line 164, in imread
    im = Image.open(name)
  File "/anaconda3/lib/python3.7/site-packages/PIL/Image.py", line 2657, in open
    % (filename if filename else fp))
OSError: cannot identify image file './data/lukea/.DS_Store'
  0%|                                                                                                                                                                                                                          | 37/300000 [12:46<2290:55:00, 27.49s/it]Process Process-10:
Traceback (most recent call last):
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 297, in _bootstrap
    self.run()
  File "/anaconda3/lib/python3.7/multiprocessing/process.py", line 99, in run
    self._target(*self._args, **self._kwargs)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 75, in initialize_batch_worker
    batch = self.get_batch(augmentor=augmentor, batch_size=batch_size)
  File "/Volumes/4tb/venv/prepare_dataset.py", line 50, in get_batch
    image = scipy.misc.imread(name=random.choice(self.dataset), mode='RGB')
  File "/anaconda3/lib/python3.7/site-packages/numpy/lib/utils.py", line 101, in newfunc
    return func(*args, **kwds)
  File "/anaconda3/lib/python3.7/site-packages/scipy/misc/pilutil.py", line 164, in imread
    im = Image.open(name)
  File "/anaconda3/lib/python3.7/site-packages/PIL/Image.py", line 2657, in open
    % (filename if filename else fp))
OSError: cannot identify image file './data/lukea/.DS_Store'
  0%|   


...