コード例 #1
0
ファイル: mock_utils.py プロジェクト: tomgillooly/geogan
    def loss_fun(mock, target, **kwargs):
        m = mock.copy()
        m.name = 'mse_loss_' + m.name

        return m
コード例 #2
0
ファイル: mock_utils.py プロジェクト: tomgillooly/geogan
def max_stub(mock, dim, keepdim):
    m = mock.copy()
    m.name += '_max'

    return m
コード例 #3
0
ファイル: mock_utils.py プロジェクト: tomgillooly/geogan
def bce_loss_stub(mock, target, **kwargs):
    m = mock.copy()
    m.name = 'bce_loss_' + m.name

    return m
コード例 #4
0
ファイル: mock_utils.py プロジェクト: tomgillooly/geogan
def softmax_stub(mock, dim):
    m = mock.copy()
    m.name = mock.name + '_softmax'

    return m