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:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user