mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 03:30:19 +01:00
Unify base class using new-style (object).
This commit is contained in:
@@ -17,7 +17,7 @@
|
||||
import tensorflow as tf
|
||||
|
||||
|
||||
class InputEmbedder:
|
||||
class InputEmbedder(object):
|
||||
def __init__(self, input_size, activation_function=tf.nn.relu, name="embedder"):
|
||||
self.name = name
|
||||
self.input_size = input_size
|
||||
|
||||
@@ -28,7 +28,7 @@ def normalized_columns_initializer(std=1.0):
|
||||
return _initializer
|
||||
|
||||
|
||||
class Head:
|
||||
class Head(object):
|
||||
def __init__(self, tuning_parameters, head_idx=0, loss_weight=1., is_local=True):
|
||||
self.head_idx = head_idx
|
||||
self.name = "head"
|
||||
|
||||
@@ -18,7 +18,7 @@ import tensorflow as tf
|
||||
import numpy as np
|
||||
|
||||
|
||||
class MiddlewareEmbedder:
|
||||
class MiddlewareEmbedder(object):
|
||||
def __init__(self, activation_function=tf.nn.relu, name="middleware_embedder"):
|
||||
self.name = name
|
||||
self.input = None
|
||||
|
||||
Reference in New Issue
Block a user