Removed relative imports
Browse files
tfing.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import tensorflow as tf
|
2 |
from tensorflow import keras
|
3 |
-
from
|
4 |
import tensorflow_probability as tfp
|
5 |
|
6 |
class TFIng(keras.Model):
|
|
|
1 |
import tensorflow as tf
|
2 |
from tensorflow import keras
|
3 |
+
from transformer import TransformerDecoder
|
4 |
import tensorflow_probability as tfp
|
5 |
|
6 |
class TFIng(keras.Model):
|
tfport.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import tensorflow as tf
|
2 |
from tensorflow import keras
|
3 |
-
from
|
4 |
|
5 |
class TFPort(keras.Model):
|
6 |
def __init__(self, crop_size, embed_dim, encoder_layers, decoder_layers, input_seq_length,
|
|
|
1 |
import tensorflow as tf
|
2 |
from tensorflow import keras
|
3 |
+
from transformer import TransformerEncoder, TransformerDecoderLayer, get_look_ahead_mask, get_padding_mask
|
4 |
|
5 |
class TFPort(keras.Model):
|
6 |
def __init__(self, crop_size, embed_dim, encoder_layers, decoder_layers, input_seq_length,
|