We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a03d7a7 commit 36f8d4bCopy full SHA for 36f8d4b
tests/unit_tests/losses/test_grpo_loss.py
@@ -4,16 +4,12 @@
4
# This source code is licensed under the BSD-style license found in the
5
# LICENSE file in the root directory of this source tree.
6
7
-import math
8
-
9
import pytest
10
import torch
11
-import torch.nn as nn
12
from forge.losses.grpo_loss import SimpleGRPOLoss
13
14
15
class TestSimpleGRPOLoss:
16
17
@pytest.fixture
18
def loss_fn(self):
19
"""Create a GRPO loss instance with default beta."""
0 commit comments