Skip to content

Commit 822d7a4

Browse files
committed
remove unuse code in gcl
1 parent e9db652 commit 822d7a4

File tree

1 file changed

+0
-21
lines changed

1 file changed

+0
-21
lines changed

ding/reward_model/guided_cost_reward_model.py

Lines changed: 0 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -13,27 +13,6 @@
1313
from .network import GCLNetwork
1414

1515

16-
class GuidedCostNN(nn.Module):
17-
18-
def __init__(
19-
self,
20-
input_size,
21-
hidden_size=128,
22-
output_size=1,
23-
):
24-
super(GuidedCostNN, self).__init__()
25-
self.net = nn.Sequential(
26-
nn.Linear(input_size, hidden_size),
27-
nn.ReLU(),
28-
nn.Linear(hidden_size, hidden_size),
29-
nn.ReLU(),
30-
nn.Linear(hidden_size, output_size),
31-
)
32-
33-
def forward(self, x):
34-
return self.net(x)
35-
36-
3716
@REWARD_MODEL_REGISTRY.register('guided_cost')
3817
class GuidedCostRewardModel(BaseRewardModel):
3918
"""

0 commit comments

Comments
 (0)