Adjust LayerNormSerializationTest test case according to #1512.
This commit is contained in:
@@ -1841,7 +1841,7 @@ BOOST_AUTO_TEST_CASE(BatchNormSerializationTest)
|
||||
*/
|
||||
BOOST_AUTO_TEST_CASE(LayerNormSerializationTest)
|
||||
{
|
||||
LayerNorm<> layer(1);
|
||||
LayerNorm<> layer(10);
|
||||
ANNLayerSerializationTest(layer);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user