Open SeekPoint opened 7 years ago
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX2 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use FMA instructions, but these are available on your machine and could speed up CPU computations.
Traceback (most recent call last):
File "train_translator.py", line 144, in
maybe I am wrong, it is report error above on tf1.0 or tf1.2
I am not sure about masked 1d, It seems that the mask operation is done on difference channels like image in RGB channel. Do you think your implementation here is in line with the paper, see pixelRNN.
git diff
diff --git a/ByteNet/model.py b/ByteNet/model.py index 4cfe3b3..3a12b5b 100644 --- a/ByteNet/model.py +++ b/ByteNet/model.py @@ -138,7 +138,7 @@ class Byte_net_model: decoder_output = self.decoder(source_embedding) loss = self.loss(decoder_output, target_sentence)
tf.summary.scalar('LOSS', loss)
@@ -220,7 +220,7 @@ class Byte_net_model:
loss = tf.nn.softmax_cross_entropy_with_logits(logits=flat_logits, labels=flat_targets, name='decoder_cross_entropy_loss')
diff --git a/train_generator.py b/train_generator.py index 78d502c..72e898b 100644