Google colab + Tensorflow: как исправить ошибку No module 'layers'?

Question

[Альберт]

Как исправить ошибку?

import tensorflow as tf
from layers import (_causal_linear, _output_linear, conv1d, dilated_conv1d)

Пишет нет модуля layers.
Насколько понимаю, layers — это что-то типа tf.layers
Запускаю Google colab в пустом проекте, python3 GPU

Источник кода: https://raw.githubusercontent.com/tomlepaine/fast-...
(3-я и 4-я строки)

Answer 1

[Сергей Тихонов]

https://programtalk.com/vs2/python/13192/fast-wave...
В оригинальном (?) примере рядом с models.py лежит layers.py, в котором все эти слои определяются.

Comments

[Альберт]

Да, ступил и не проверил файл layers, хотя это сделать было бы логично.
Сбили с толку пути.

все работает, таким образом:
Первая ячейка

!git clone https://github.com/tomlepaine/fast-wavenet.git
%cd fast-wavenet/wavenet

А не (просто %cd fast-wavenet) как делал раньше
Тогда вторая ячейка (если использовать демо ipynb пример) будет такой:

from time import time
from utils import make_batch
from models import Model, Generator
from IPython.display import Audio
%matplotlib inline