From ba7c69ecba8130b5470826b4ecd1b7478c25bff1 Mon Sep 17 00:00:00 2001 From: Lars Gregori Date: Sun, 9 Apr 2017 13:47:24 +0200 Subject: [PATCH] Change imports to use sonnet; same as most other files --- preprocess.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/preprocess.py b/preprocess.py index a08fffd..038e66a 100644 --- a/preprocess.py +++ b/preprocess.py @@ -19,12 +19,11 @@ from __future__ import print_function import numpy as np +import sonnet as snt import tensorflow as tf -import nn - -class Clamp(nn.AbstractModule): +class Clamp(snt.AbstractModule): def __init__(self, min_value=None, max_value=None, name="clamp"): super(Clamp, self).__init__(name=name) @@ -40,7 +39,7 @@ def _build(self, inputs): return output -class LogAndSign(nn.AbstractModule): +class LogAndSign(snt.AbstractModule): """Log and sign preprocessing. As described in https://arxiv.org/pdf/1606.04474v1.pdf (Appendix A).