Skip to content

Commit d5bbebb

Browse files
committed
Fix deprecation warning
1 parent 8d3cab2 commit d5bbebb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

plasma/models/builder.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,11 @@
1010
Input,
1111
Dense, Activation, Dropout, Lambda,
1212
Reshape, Flatten, Permute, # RepeatVector
13-
LSTM, CuDNNLSTM, SimpleRNN, BatchNormalization,
13+
LSTM, SimpleRNN, BatchNormalization,
1414
Convolution1D, MaxPooling1D, TimeDistributed,
1515
Concatenate
1616
)
17+
from tensorflow.compat.v1.keras.layers import CuDNNLSTM
1718
from tensorflow.keras.callbacks import Callback
1819
from tensorflow.keras.regularizers import l2 # l1, l1_l2
1920

0 commit comments

Comments
 (0)