We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4584ea9 commit 0e3e1aeCopy full SHA for 0e3e1ae
1 file changed
train.py
@@ -1,5 +1,4 @@
1
#!/usr/bin/env python
2
-from utils import train_utils, googlenet_load
3
import json
4
import tensorflow.contrib.slim as slim
5
import datetime
@@ -22,6 +21,8 @@
22
21
random.seed(0)
23
np.random.seed(0)
24
+from utils import train_utils, googlenet_load
25
+
26
@ops.RegisterGradient("Hungarian")
27
def _hungarian_grad(op, *args):
28
return map(array_ops.zeros_like, op.inputs)
0 commit comments