Skip to content

Commit e4b530c

Browse files
author
Niklas Gustafsson
committed
Fixed a unit test.
1 parent 18b3298 commit e4b530c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/TorchSharpTest/TestAutogradFunction.cs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -162,15 +162,15 @@ public void TestCustomTwoInputOneGradientLinearFunction_CUDA()
162162
public void TestCustomLinearXORLearn_CPU()
163163
{
164164
float loss = TrainXOR(torch.CPU);
165-
LossIsClose(loss, 0.4513f);
165+
LossIsClose(0.4513f, loss);
166166
}
167167

168168
[Fact]
169169
public void TestCustomLinearXORLearn_CUDA()
170170
{
171171
if (torch.cuda.is_available()) {
172172
float loss = TrainXOR(torch.CUDA);
173-
LossIsClose(loss, 0.4513f);
173+
LossIsClose(0.4513f, loss);
174174
}
175175
}
176176

@@ -179,7 +179,7 @@ public void TestCustomLinearXORLearn_Parallel_CPU()
179179
{
180180
Enumerable.Range(0, 20).AsParallel().ForAll(i => {
181181
float loss = TrainXOR(torch.CPU);
182-
LossIsClose(loss, 0.4513f);
182+
LossIsClose(0.4513f, loss);
183183
});
184184
}
185185

0 commit comments

Comments
 (0)