This repository was archived by the owner on Apr 29, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +16
-0
lines changed
kilroy_module_pytorch_py_sdk/src/kilroy_module_pytorch_py_sdk/modules Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Original file line number Diff line number Diff line change @@ -62,6 +62,10 @@ def name(cls) -> str:
62
62
def label (cls ) -> str :
63
63
return "Supervised Loss"
64
64
65
+ @classproperty
66
+ def group (cls ) -> str :
67
+ return "1"
68
+
65
69
@classproperty
66
70
def config (cls ) -> Dict [str , Any ]:
67
71
return {
@@ -86,6 +90,10 @@ def name(cls) -> str:
86
90
def label (cls ) -> str :
87
91
return "Reinforced Score"
88
92
93
+ @classproperty
94
+ def group (cls ) -> str :
95
+ return "1"
96
+
89
97
@classproperty
90
98
def config (cls ) -> Dict [str , Any ]:
91
99
return {
@@ -110,6 +118,10 @@ def name(cls) -> str:
110
118
def label (cls ) -> str :
111
119
return "Reward Model Loss"
112
120
121
+ @classproperty
122
+ def group (cls ) -> str :
123
+ return "1"
124
+
113
125
@classproperty
114
126
def config (cls ) -> Dict [str , Any ]:
115
127
return {
@@ -134,6 +146,10 @@ def name(cls) -> str:
134
146
def label (cls ) -> str :
135
147
return "Reward Model Score"
136
148
149
+ @classproperty
150
+ def group (cls ) -> str :
151
+ return "2"
152
+
137
153
@classproperty
138
154
def config (cls ) -> Dict [str , Any ]:
139
155
return {
You can’t perform that action at this time.
0 commit comments