1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

Unify base class using new-style (object).

This commit is contained in:
cxx
2017-10-26 10:16:11 +08:00
committed by Itai Caspi
parent 39cf78074c
commit f43c951c2d
16 changed files with 28 additions and 28 deletions

View File

@@ -22,7 +22,7 @@ from utils import force_list
from architectures.neon_components.losses import *
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"