Skip to content

Commit 863cc65

Browse files
authored
fix typos and memory optimisation
1 parent 4d164ea commit 863cc65

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tensorlayer/layers.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1888,7 +1888,10 @@ def __init__(
18881888
Layer.__init__(self, name=name)
18891889
self.inputs = layer.outputs
18901890
self.offset_layer = offset_layer
1891-
1891+
1892+
if tf.__version__ < "1.4":
1893+
raise Exception("Deformable CNN layer requires tensrflow 1.4 or higher version")
1894+
18921895
print(" [TL] DeformableConv2dLayer %s: shape:%s, act:%s" %
18931896
(self.name, str(shape), act.__name__))
18941897

@@ -6990,6 +6993,7 @@ def get_batch(self, data, bucket_id, PAD_ID=0, GO_ID=1, EOS_ID=2, UNK_ID=3):
69906993

69916994

69926995

6996+
69936997

69946998

69956999
#

0 commit comments

Comments
 (0)