File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -162,15 +162,15 @@ public void TestCustomTwoInputOneGradientLinearFunction_CUDA()
162
162
public void TestCustomLinearXORLearn_CPU ( )
163
163
{
164
164
float loss = TrainXOR ( torch . CPU ) ;
165
- LossIsClose ( loss , 0.4513f ) ;
165
+ LossIsClose ( 0.4513f , loss ) ;
166
166
}
167
167
168
168
[ Fact ]
169
169
public void TestCustomLinearXORLearn_CUDA ( )
170
170
{
171
171
if ( torch . cuda . is_available ( ) ) {
172
172
float loss = TrainXOR ( torch . CUDA ) ;
173
- LossIsClose ( loss , 0.4513f ) ;
173
+ LossIsClose ( 0.4513f , loss ) ;
174
174
}
175
175
}
176
176
@@ -179,7 +179,7 @@ public void TestCustomLinearXORLearn_Parallel_CPU()
179
179
{
180
180
Enumerable . Range ( 0 , 20 ) . AsParallel ( ) . ForAll ( i => {
181
181
float loss = TrainXOR ( torch . CPU ) ;
182
- LossIsClose ( loss , 0.4513f ) ;
182
+ LossIsClose ( 0.4513f , loss ) ;
183
183
} ) ;
184
184
}
185
185
You can’t perform that action at this time.
0 commit comments