复现的yolox的代码在本地能正常运行，复制到BML上却运行中间报错

[Wei-JL]

bug描述 Describe the Bug

File "/home/aistudio/yolox_paddle/nets/yolo_training.py", line 228, in get_losses
reg_targets = paddle.concat(reg_targets, 0)
File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/tensor/manipulation.py", line 345, in concat
return paddle.fluid.layers.concat(input=x, axis=axis, name=name)
File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/layers/tensor.py", line 327, in concat
return _C_ops.concat(input, 'axis', axis)
ValueError: (InvalidArgument) The shape of input[0] and input[10] is expected to be equal.But received input[0]'s shape = [171, 4], input[10]'s shape = [4].
[Hint: Expected inputs_dims[i].size() == out_dims.size(), but received inputs_dims[i].size():1 != out_dims.size():2.] (at /paddle/paddle/fluid/operators/concat_op.h:40)
[operator < concat > error]
是一个简单的concat，reg_targets这个是一个list，里面装了很多tensor。

其他补充信息 Additional Supplementary Information

reg_targets = [Tensor(shape=[171, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.68125153, 0. , 0. ], [0. , 0.61638886, 0. , 0. ], [0. , 0.56190431, 0. , 0. ], [0. , 0.43946189, 0. , 0. ], [0. , 0.44488573, 0. , 0. ], [0. , 0.47852728, 0. , 0. ], [0. , 0.62957996, 0. , 0. ], [0. , 0.43927443, 0. , 0. ], [0. , 0.60877669, 0. , 0. ], [0. , 0.55816185, 0. , 0. ], [0. , 0.56835550, 0. , 0. ], [0. , 0.63362104, 0. , 0. ], [0. , 0.61195499, 0. , 0. ], [0. , 0.60083026, 0. , 0. ], [0. , 0.69176883, 0. , 0. ], [0. , 0.78723443, 0. , 0. ], [0. , 0.66117030, 0. , 0. ], [0. , 0.48903719, 0. , 0. ], [0. , 0.75129575, 0. , 0. ], [0. , 0.51143688, 0. , 0. ], [0. , 0.52383155, 0. , 0. ], [0. , 0.70052105, 0. , 0. ], [0. , 0.67497063, 0. , 0. ], [0. , 0.60524029, 0. , 0. ], [0. , 0.62612706, 0. , 0. ], [0. , 0.55234683, 0. , 0. ], [0. , 0.68246895, 0. , 0. ], [0. , 0.80535686, 0. , 0. ], [0. , 0.74578601, 0. , 0. ], [0. , 0.59614158, 0. , 0. ], [0. , 0.64921761, 0. , 0. ], [0. , 0.57915759, 0. , 0. ], [0. , 0.63296276, 0. , 0. ], [0. , 0.58831900, 0. , 0. ], [0. , 0.64894462, 0. , 0. ], [0. , 0.59109133, 0. , 0. ], [0. , 0.56036633, 0. , 0. ], [0. , 0.54420942, 0. , 0. ], [0. , 0.64433831, 0. , 0. ], [0. , 0.61904037, 0. , 0. ], [0. , 0.54237092, 0. , 0. ], [0. , 0.64823383, 0. , 0. ], [0. , 0.64586020, 0. , 0. ], [0. , 0.63307267, 0. , 0. ], [0. , 0.60748333, 0. , 0. ], [0. , 0.65849137, 0. , 0. ], [0. , 0.65190238, 0. , 0. ], [0. , 0.55516768, 0. , 0. ], [0. , 0.63637078, 0. , 0. ], [0. , 0.61408645, 0. , 0. ], [0. , 0.62407315, 0. , 0. ], [0. , 0.63053858, 0. , 0. ], [0. , 0.67593896, 0. , 0. ], [0. , 0.65765327, 0. , 0. ], [0. , 0.69886941, 0. , 0. ], [0. , 0.69492429, 0. , 0. ], [0. , 0.51320589, 0. , 0. ], [0. , 0.49330410, 0. , 0. ], [0. , 0.60048628, 0. , 0. ], [0. , 0.64188772, 0. , 0. ], [0. , 0.58314878, 0. , 0. ], [0. , 0.67104393, 0. , 0. ], [0. , 0.53367102, 0. , 0. ], [0. , 0.56394446, 0. , 0. ], [0. , 0.72208142, 0. , 0. ], [0. , 0.77083242, 0. , 0. ], [0. , 0.73275447, 0. , 0. ], [0. , 0.72078317, 0. , 0. ], [0. , 0.52254206, 0. , 0. ], [0. , 0.60149235, 0. , 0. ], [0. , 0.63276452, 0. , 0. ], [0. , 0.53549421, 0. , 0. ], [0. , 0.80267149, 0. , 0. ], [0. , 0.76598090, 0. , 0. ], [0. , 0.77543324, 0. , 0. ], [0. , 0.68687046, 0. , 0. ], [0. , 0.77459800, 0. , 0. ], [0. , 0.73950213, 0. , 0. ], [0. , 0.61742848, 0. , 0. ], [0. , 0.58835167, 0. , 0. ], [0. , 0.60286683, 0. , 0. ], [0. , 0.67223644, 0. , 0. ], [0. , 0.68155295, 0. , 0. ], [0. , 0.69334382, 0. , 0. ], [0. , 0.75476438, 0. , 0. ], [0. , 0.75891566, 0. , 0. ], [0. , 0.62486058, 0. , 0. ], [0. , 0.64384836, 0. , 0. ], [0. , 0.66317248, 0. , 0. ], [0. , 0.68622339, 0. , 0. ], [0. , 0.69253522, 0. , 0. ], [0. , 0.75637400, 0. , 0. ], [0. , 0.72031492, 0. , 0. ], [0. , 0.67405766, 0. , 0. ], [0. , 0.66381735, 0. , 0. ], [0. , 0.63460374, 0. , 0. ], [0. , 0.60107923, 0. , 0. ], [0. , 0.73606706, 0. , 0. ], [0. , 0.73520637, 0. , 0. ], [0. , 0.76867861, 0. , 0. ], [0. , 0.77327883, 0. , 0. ], [0. , 0.61933762, 0. , 0. ], [0. , 0.62543273, 0. , 0. ], [0. , 0.60498482, 0. , 0. ], [0. , 0.73971164, 0. , 0. ], [0. , 0.42505515, 0. , 0. ], [0. , 0.44445568, 0. , 0. ], [0. , 0.60151190, 0. , 0. ], [0. , 0.79700255, 0. , 0. ], [0. , 0.73372090, 0. , 0. ], [0. , 0.74412698, 0. , 0. ], [0. , 0.88653731, 0. , 0. ], [0. , 0.77871388, 0. , 0. ], [0. , 0.80660200, 0. , 0. ], [0. , 0.42322996, 0. , 0. ], [0. , 0.82249719, 0. , 0. ], [0. , 0.75249451, 0. , 0. ], [0. , 0.73284072, 0. , 0. ], [0. , 0.82844412, 0. , 0. ], [0. , 0.76153058, 0. , 0. ], [0. , 0.75680453, 0. , 0. ], [0. , 0.69528145, 0. , 0. ], [0. , 0.77045482, 0. , 0. ], [0. , 0.75635242, 0. , 0. ], [0. , 0.67338306, 0. , 0. ], [0. , 0.59286487, 0. , 0. ], [0. , 0.58222431, 0. , 0. ], [0. , 0.72766542, 0. , 0. ], [0. , 0.66793680, 0. , 0. ], [0. , 0.68275130, 0. , 0. ], [0. , 0.66293180, 0. , 0. ], [0. , 0.63750237, 0. , 0. ], [0. , 0.53421897, 0. , 0. ], [0. , 0.56787944, 0. , 0. ], [0. , 0.72638029, 0. , 0. ], [0. , 0.76640213, 0. , 0. ], [0. , 0.71980560, 0. , 0. ], [0. , 0.53394896, 0. , 0. ], [0. , 0.58185273, 0. , 0. ], [0. , 0.53827053, 0. , 0. ], [0. , 0.70164371, 0. , 0. ], [0. , 0.67156750, 0. , 0. ], [0. , 0.79283649, 0. , 0. ], [0. , 0.73807824, 0. , 0. ], [0. , 0.85937351, 0. , 0. ], [0. , 0.91270000, 0. , 0. ], [0. , 0.85017616, 0. , 0. ], [0. , 0.84660941, 0. , 0. ], [0. , 0.74836284, 0. , 0. ], [0. , 0.73819637, 0. , 0. ], [0. , 0.60868847, 0. , 0. ], [0. , 0.87053525, 0. , 0. ], [0. , 0.85458022, 0. , 0. ], [0. , 0.71529800, 0. , 0. ], [0. , 0.67534274, 0. , 0. ], [0. , 0.72497356, 0. , 0. ], [0. , 0.64705181, 0. , 0. ], [0. , 0.66793483, 0. , 0. ], [0. , 0.67014462, 0. , 0. ], [0. , 0.68263167, 0. , 0. ], [0. , 0.70062780, 0. , 0. ], [0. , 0.72549486, 0. , 0. ], [0. , 0.76152796, 0. , 0. ], [0. , 0.65310597, 0. , 0. ], [0. , 0.77667296, 0. , 0. ], [0. , 0.73229605, 0. , 0. ], [0. , 0.71225494, 0. , 0. ], [0. , 0.82073200, 0. , 0. ], [0. , 0.77869880, 0. , 0. ], [0. , 0.82137996, 0. , 0. ], [0. , 0.81569833, 0. , 0. ]]), Tensor(shape=[14, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.06948655, 0. , 0. ], [0. , 0.06191347, 0. , 0. ], [0. , 0.09574467, 0. , 0. ], [0. , 0.08406589, 0. , 0. ], [0. , 0.18006884, 0. , 0. ], [0. , 0.14306632, 0. , 0. ], [0. , 0.23651247, 0. , 0. ], [0. , 0.02021633, 0. , 0. ], [0. , 0.00901831, 0. , 0. ], [0. , 0.03427640, 0. , 0. ], [0. , 0.03910261, 0. , 0. ], [0. , 0.02501057, 0. , 0. ], [0. , 0.00851180, 0. , 0. ], [0. , 0.00151898, 0. , 0. ]]), Tensor(shape=[5, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0. , 0.57751799], [0. , 0. , 0. , 0.85893720], [0. , 0. , 0. , 0.74348867], [0. , 0. , 0. , 0.59163076], [0. , 0. , 0. , 0.54832727]]), Tensor(shape=[5, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0.53447074, 0. ], [0. , 0. , 0.54460305, 0. ], [0. , 0. , 0.58602029, 0. ], [0. , 0. , 0.57000816, 0. ], [0. , 0. , 0.60871691, 0. ]]), Tensor(shape=[14, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.72217453, 0. , 0. ], [0. , 0.80119288, 0. , 0. ], [0. , 0.81825382, 0. , 0. ], [0. , 0.80139160, 0. , 0. ], [0. , 0.78544897, 0. , 0. ], [0. , 0.84075105, 0. , 0. ], [0. , 0.84032732, 0. , 0. ], [0. , 0.73084033, 0. , 0. ], [0. , 0.75326484, 0. , 0. ], [0. , 0.72907597, 0. , 0. ], [0. , 0.80680728, 0. , 0. ], [0. , 0.68591100, 0. , 0. ], [0. , 0.70849496, 0. , 0. ], [0. , 0.76388431, 0. , 0. ]]), Tensor(shape=[46, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.79141819, 0. , 0. ], [0. , 0.77055949, 0. , 0. ], [0. , 0.62456048, 0. , 0. ], [0. , 0.72702444, 0. , 0. ], [0. , 0.63975555, 0. , 0. ], [0. , 0.67698288, 0. , 0. ], [0. , 0.72322839, 0. , 0. ], [0. , 0.64822644, 0. , 0. ], [0. , 0.64903849, 0. , 0. ], [0. , 0.69396472, 0. , 0. ], [0. , 0.64928627, 0. , 0. ], [0. , 0.69066584, 0. , 0. ], [0. , 0.73675996, 0. , 0. ], [0. , 0.82784629, 0. , 0. ], [0. , 0.63949221, 0. , 0. ], [0. , 0.65694171, 0. , 0. ], [0. , 0.63992214, 0. , 0. ], [0. , 0.61650109, 0. , 0. ], [0. , 0.81746215, 0. , 0. ], [0. , 0.73146749, 0. , 0. ], [0. , 0.77986497, 0. , 0. ], [0. , 0.89348143, 0. , 0. ], [0. , 0.79339820, 0. , 0. ], [0. , 0.79347605, 0. , 0. ], [0. , 0.75590020, 0. , 0. ], [0. , 0.76081800, 0. , 0. ], [0. , 0.78214544, 0. , 0. ], [0. , 0.78060442, 0. , 0. ], [0. , 0.83873826, 0. , 0. ], [0. , 0.74698287, 0. , 0. ], [0. , 0.72593075, 0. , 0. ], [0. , 0.67426383, 0. , 0. ], [0. , 0.69063550, 0. , 0. ], [0. , 0.81124312, 0. , 0. ], [0. , 0.79459381, 0. , 0. ], [0. , 0.76853001, 0. , 0. ], [0. , 0.79654318, 0. , 0. ], [0. , 0.71337020, 0. , 0. ], [0. , 0.72516859, 0. , 0. ], [0. , 0.68900245, 0. , 0. ], [0. , 0.68302894, 0. , 0. ], [0. , 0.73089826, 0. , 0. ], [0. , 0.62145430, 0. , 0. ], [0. , 0.68094754, 0. , 0. ], [0. , 0.58134151, 0. , 0. ], [0. , 0.74927145, 0. , 0. ]]), Tensor(shape=[37, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.70602167, 0. , 0. ], [0. , 0.70122302, 0. , 0. ], [0. , 0.71185136, 0. , 0. ], [0. , 0.69922471, 0. , 0. ], [0. , 0.68759197, 0. , 0. ], [0. , 0.66191208, 0. , 0. ], [0. , 0.64629769, 0. , 0. ], [0. , 0.03515900, 0. , 0. ], [0. , 0.69809264, 0. , 0. ], [0. , 0.74643111, 0. , 0. ], [0. , 0.80376011, 0. , 0. ], [0. , 0.63317871, 0. , 0. ], [0. , 0.56874996, 0. , 0. ], [0. , 0.80328381, 0. , 0. ], [0. , 0.84120625, 0. , 0. ], [0. , 0.85890234, 0. , 0. ], [0. , 0.67154366, 0. , 0. ], [0. , 0.64948541, 0. , 0. ], [0. , 0.78308988, 0. , 0. ], [0. , 0.78900820, 0. , 0. ], [0. , 0.80552137, 0. , 0. ], [0. , 0.81758082, 0. , 0. ], [0. , 0.76802862, 0. , 0. ], [0. , 0.71966797, 0. , 0. ], [0. , 0.72767287, 0. , 0. ], [0. , 0.75674313, 0. , 0. ], [0. , 0.65683198, 0. , 0. ], [0. , 0.66208571, 0. , 0. ], [0. , 0.83951604, 0. , 0. ], [0. , 0.79015630, 0. , 0. ], [0. , 0.87050259, 0. , 0. ], [0. , 0.82193553, 0. , 0. ], [0. , 0.81733537, 0. , 0. ], [0. , 0.80431467, 0. , 0. ], [0. , 0.77698249, 0. , 0. ], [0. , 0.84485221, 0. , 0. ], [0. , 0.76861632, 0. , 0. ]]), Tensor(shape=[5, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0. , 0.56235188], [0. , 0. , 0. , 0.56762308], [0. , 0. , 0. , 0.63796055], [0. , 0. , 0. , 0.60854864], [0. , 0. , 0. , 0.54621887]]), Tensor(shape=[5, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0.65150052, 0. ], [0. , 0. , 0.66785794, 0. ], [0. , 0. , 0.64549458, 0. ], [0. , 0. , 0.80208284, 0. ], [0. , 0. , 0.60254604, 0. ]]), Tensor(shape=[123, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.64725512, 0. , 0. ], [0. , 0.46336845, 0. , 0. ], [0. , 0.80802745, 0. , 0. ], [0. , 0.82008928, 0. , 0. ], [0. , 0.89079875, 0. , 0. ], [0. , 0.78063327, 0. , 0. ], [0. , 0.80120325, 0. , 0. ], [0. , 0.80964041, 0. , 0. ], [0. , 0.84132987, 0. , 0. ], [0. , 0.89166456, 0. , 0. ], [0. , 0.83730739, 0. , 0. ], [0. , 0.84077275, 0. , 0. ], [0. , 0.82260215, 0. , 0. ], [0. , 0.87591028, 0. , 0. ], [0. , 0.80767471, 0. , 0. ], [0. , 0.81437224, 0. , 0. ], [0. , 0.71002960, 0. , 0. ], [0. , 0.73870552, 0. , 0. ], [0. , 0.81524938, 0. , 0. ], [0. , 0.78222048, 0. , 0. ], [0. , 0.76731855, 0. , 0. ], [0. , 0.79824466, 0. , 0. ], [0. , 0.65949512, 0. , 0. ], [0. , 0.67319465, 0. , 0. ], [0. , 0.76176584, 0. , 0. ], [0. , 0.74836791, 0. , 0. ], [0. , 0.84746081, 0. , 0. ], [0. , 0.81161320, 0. , 0. ], [0. , 0.83235180, 0. , 0. ], [0. , 0.75655532, 0. , 0. ], [0. , 0.84336704, 0. , 0. ], [0. , 0.86710888, 0. , 0. ], [0. , 0.87286586, 0. , 0. ], [0. , 0.88154835, 0. , 0. ], [0. , 0.81185657, 0. , 0. ], [0. , 0.72274441, 0. , 0. ], [0. , 0.76822668, 0. , 0. ], [0. , 0.75752902, 0. , 0. ], [0. , 0.77117896, 0. , 0. ], [0. , 0.68277848, 0. , 0. ], [0. , 0.68622696, 0. , 0. ], [0. , 0.67113227, 0. , 0. ], [0. , 0.69597185, 0. , 0. ], [0. , 0.65799522, 0. , 0. ], [0. , 0.67250758, 0. , 0. ], [0. , 0.66222757, 0. , 0. ], [0. , 0.84852779, 0. , 0. ], [0. , 0.81286204, 0. , 0. ], [0. , 0.78233498, 0. , 0. ], [0. , 0.75141031, 0. , 0. ], [0. , 0.74184501, 0. , 0. ], [0. , 0.66685838, 0. , 0. ], [0. , 0.71196246, 0. , 0. ], [0. , 0.80475920, 0. , 0. ], [0. , 0.79096615, 0. , 0. ], [0. , 0.79750359, 0. , 0. ], [0. , 0.88780689, 0. , 0. ], [0. , 0.78856325, 0. , 0. ], [0. , 0.79156107, 0. , 0. ], [0. , 0.87763256, 0. , 0. ], [0. , 0.82952553, 0. , 0. ], [0. , 0.84784752, 0. , 0. ], [0. , 0.93173784, 0. , 0. ], [0. , 0.83349168, 0. , 0. ], [0. , 0.80533230, 0. , 0. ], [0. , 0.84375972, 0. , 0. ], [0. , 0.82756048, 0. , 0. ], [0. , 0.66119093, 0. , 0. ], [0. , 0.67767191, 0. , 0. ], [0. , 0.67515743, 0. , 0. ], [0. , 0.71111572, 0. , 0. ], [0. , 0.66159689, 0. , 0. ], [0. , 0.75536752, 0. , 0. ], [0. , 0.75967407, 0. , 0. ], [0. , 0.64139384, 0. , 0. ], [0. , 0.68696952, 0. , 0. ], [0. , 0.64148331, 0. , 0. ], [0. , 0.67471915, 0. , 0. ], [0. , 0.66606551, 0. , 0. ], [0. , 0.75377667, 0. , 0. ], [0. , 0.66420543, 0. , 0. ], [0. , 0.67672545, 0. , 0. ], [0. , 0.62953341, 0. , 0. ], [0. , 0.57110459, 0. , 0. ], [0. , 0.82834584, 0. , 0. ], [0. , 0.81199646, 0. , 0. ], [0. , 0.75671262, 0. , 0. ], [0. , 0.79527122, 0. , 0. ], [0. , 0.63237607, 0. , 0. ], [0. , 0.79204333, 0. , 0. ], [0. , 0.77075899, 0. , 0. ], [0. , 0.86810052, 0. , 0. ], [0. , 0.64262289, 0. , 0. ], [0. , 0.57133716, 0. , 0. ], [0. , 0.56755704, 0. , 0. ], [0. , 0.60647231, 0. , 0. ], [0. , 0.79782051, 0. , 0. ], [0. , 0.71789175, 0. , 0. ], [0. , 0.65727884, 0. , 0. ], [0. , 0.72232872, 0. , 0. ], [0. , 0.78039414, 0. , 0. ], [0. , 0.76118439, 0. , 0. ], [0. , 0.78804272, 0. , 0. ], [0. , 0.75355065, 0. , 0. ], [0. , 0.88168579, 0. , 0. ], [0. , 0.77767593, 0. , 0. ], [0. , 0.72730792, 0. , 0. ], [0. , 0.71466166, 0. , 0. ], [0. , 0.66699392, 0. , 0. ], [0. , 0.79436678, 0. , 0. ], [0. , 0.72628200, 0. , 0. ], [0. , 0.69732875, 0. , 0. ], [0. , 0.80646753, 0. , 0. ], [0. , 0.82028598, 0. , 0. ], [0. , 0.81060046, 0. , 0. ], [0. , 0.68557006, 0. , 0. ], [0. , 0.70195323, 0. , 0. ], [0. , 0.66626889, 0. , 0. ], [0. , 0.71337187, 0. , 0. ], [0. , 0.02387712, 0. , 0. ], [0. , 0.26703164, 0. , 0. ], [0. , 0.15853418, 0. , 0. ], [0. , 0.87909019, 0. , 0. ]]), Tensor(shape=[1, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.06278976, 0. , 0. ]]), Tensor(shape=[117, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.53364277, 0. , 0. ], [0. , 0.71536082, 0. , 0. ], [0. , 0.58132297, 0. , 0. ], [0. , 0.55981249, 0. , 0. ], [0. , 0.72996843, 0. , 0. ], [0. , 0.60135293, 0. , 0. ], [0. , 0.81359613, 0. , 0. ], [0. , 0.84161997, 0. , 0. ], [0. , 0.79172796, 0. , 0. ], [0. , 0.66244757, 0. , 0. ], [0. , 0.63579816, 0. , 0. ], [0. , 0.71408629, 0. , 0. ], [0. , 0.91207063, 0. , 0. ], [0. , 0.68797755, 0. , 0. ], [0. , 0.62276191, 0. , 0. ], [0. , 0.72797716, 0. , 0. ], [0. , 0.61759597, 0. , 0. ], [0. , 0.59851652, 0. , 0. ], [0. , 0.55620116, 0. , 0. ], [0. , 0.59247005, 0. , 0. ], [0. , 0.62608218, 0. , 0. ], [0. , 0.58560318, 0. , 0. ], [0. , 0.52909046, 0. , 0. ], [0. , 0.54474336, 0. , 0. ], [0. , 0.54815054, 0. , 0. ], [0. , 0.64726800, 0. , 0. ], [0. , 0.56352472, 0. , 0. ], [0. , 0.64501101, 0. , 0. ], [0. , 0.78214926, 0. , 0. ], [0. , 0.65965539, 0. , 0. ], [0. , 0.61015397, 0. , 0. ], [0. , 0.69931692, 0. , 0. ], [0. , 0.52216840, 0. , 0. ], [0. , 0.57788414, 0. , 0. ], [0. , 0.53386837, 0. , 0. ], [0. , 0.60223377, 0. , 0. ], [0. , 0.51628965, 0. , 0. ], [0. , 0.54172748, 0. , 0. ], [0. , 0.80310702, 0. , 0. ], [0. , 0.71816653, 0. , 0. ], [0. , 0.80106688, 0. , 0. ], [0. , 0.72267282, 0. , 0. ], [0. , 0.76473457, 0. , 0. ], [0. , 0.80302370, 0. , 0. ], [0. , 0.80121994, 0. , 0. ], [0. , 0.81360620, 0. , 0. ], [0. , 0.78484267, 0. , 0. ], [0. , 0.79999089, 0. , 0. ], [0. , 0.82587481, 0. , 0. ], [0. , 0.75301361, 0. , 0. ], [0. , 0.77839661, 0. , 0. ], [0. , 0.87540466, 0. , 0. ], [0. , 0.76760405, 0. , 0. ], [0. , 0.82457072, 0. , 0. ], [0. , 0.84905595, 0. , 0. ], [0. , 0.87475795, 0. , 0. ], [0. , 0.75163800, 0. , 0. ], [0. , 0.78545231, 0. , 0. ], [0. , 0.85414076, 0. , 0. ], [0. , 0.87504131, 0. , 0. ], [0. , 0.71033710, 0. , 0. ], [0. , 0.88523060, 0. , 0. ], [0. , 0.54568499, 0. , 0. ], [0. , 0.83278239, 0. , 0. ], [0. , 0.86449069, 0. , 0. ], [0. , 0.62666333, 0. , 0. ], [0. , 0.66958338, 0. , 0. ], [0. , 0.70226425, 0. , 0. ], [0. , 0.71075821, 0. , 0. ], [0. , 0.88220197, 0. , 0. ], [0. , 0.78040367, 0. , 0. ], [0. , 0.67402285, 0. , 0. ], [0. , 0.70217979, 0. , 0. ], [0. , 0.81057328, 0. , 0. ], [0. , 0.82979399, 0. , 0. ], [0. , 0.81673628, 0. , 0. ], [0. , 0.82598341, 0. , 0. ], [0. , 0.75532693, 0. , 0. ], [0. , 0.80156428, 0. , 0. ], [0. , 0.90173501, 0. , 0. ], [0. , 0.66682142, 0. , 0. ], [0. , 0.63877660, 0. , 0. ], [0. , 0.64759034, 0. , 0. ], [0. , 0.63920981, 0. , 0. ], [0. , 0.61627173, 0. , 0. ], [0. , 0.60430580, 0. , 0. ], [0. , 0.82372659, 0. , 0. ], [0. , 0.79303050, 0. , 0. ], [0. , 0.84053361, 0. , 0. ], [0. , 0.75587803, 0. , 0. ], [0. , 0.76138747, 0. , 0. ], [0. , 0.82797348, 0. , 0. ], [0. , 0.81992197, 0. , 0. ], [0. , 0.68778843, 0. , 0. ], [0. , 0.78526074, 0. , 0. ], [0. , 0.68780184, 0. , 0. ], [0. , 0.71181864, 0. , 0. ], [0. , 0.82110083, 0. , 0. ], [0. , 0.73331237, 0. , 0. ], [0. , 0.75381804, 0. , 0. ], [0. , 0.67658252, 0. , 0. ], [0. , 0.77052957, 0. , 0. ], [0. , 0.80756354, 0. , 0. ], [0. , 0.72633857, 0. , 0. ], [0. , 0.86250883, 0. , 0. ], [0. , 0.80859107, 0. , 0. ], [0. , 0.78337210, 0. , 0. ], [0. , 0.62059474, 0. , 0. ], [0. , 0.78127325, 0. , 0. ], [0. , 0.66878515, 0. , 0. ], [0. , 0.66392916, 0. , 0. ], [0. , 0.60990781, 0. , 0. ], [0. , 0.77111566, 0. , 0. ], [0. , 0.72845358, 0. , 0. ], [0. , 0.05277224, 0. , 0. ], [0. , 0.07514095, 0. , 0. ], [0. , 0.06167213, 0. , 0. ]]), Tensor(shape=[0, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, []), Tensor(shape=[5, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0.52165759, 0. ], [0. , 0. , 0.63168722, 0. ], [0. , 0. , 0.56243467, 0. ], [0. , 0. , 0.54787862, 0. ], [0. , 0. , 0.57190073, 0. ]]), Tensor(shape=[3, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0.33553368, 0. ], [0. , 0. , 0.41371459, 0. ], [0. , 0. , 0.39249837, 0. ]]), Tensor(shape=[24, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0.50337917, 0. , 0. , 0. ], [0.52705061, 0. , 0. , 0. ], [0.60682625, 0. , 0. , 0. ], [0.65174520, 0. , 0. , 0. ], [0.73544121, 0. , 0. , 0. ], [0.80019337, 0. , 0. , 0. ], [0.63625342, 0. , 0. , 0. ], [0.69553363, 0. , 0. , 0. ], [0.65721506, 0. , 0. , 0. ], [0.84587282, 0. , 0. , 0. ], [0.47371435, 0. , 0. , 0. ], [0.43661970, 0. , 0. , 0. ], [0.63080543, 0. , 0. , 0. ], [0.58398885, 0. , 0. , 0. ], [0.52820623, 0. , 0. , 0. ], [0.61072403, 0. , 0. , 0. ], [0.65209717, 0. , 0. , 0. ], [0.85791224, 0. , 0. , 0. ], [0.51862520, 0. , 0. , 0. ], [0.79741704, 0. , 0. , 0. ], [0.56297123, 0. , 0. , 0. ], [0.67166603, 0. , 0. , 0. ], [0.85879898, 0. , 0. , 0. ], [0.67631608, 0. , 0. , 0. ]]), Tensor(shape=[2, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.19848906, 0. , 0. ], [0. , 0.21148634, 0. , 0. ]]), Tensor(shape=[43, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.92462981, 0. , 0. ], [0. , 0.78808218, 0. , 0. ], [0. , 0.74158484, 0. , 0. ], [0. , 0.62841648, 0. , 0. ], [0. , 0.70283729, 0. , 0. ], [0. , 0.65540725, 0. , 0. ], [0. , 0.83427280, 0. , 0. ], [0. , 0.80941319, 0. , 0. ], [0. , 0.87251168, 0. , 0. ], [0. , 0.83515215, 0. , 0. ], [0. , 0.81256598, 0. , 0. ], [0. , 0.76535499, 0. , 0. ], [0. , 0.80377847, 0. , 0. ], [0. , 0.82421005, 0. , 0. ], [0. , 0.83123100, 0. , 0. ], [0. , 0.61430579, 0. , 0. ], [0. , 0.63737327, 0. , 0. ], [0. , 0.74329454, 0. , 0. ], [0. , 0.79054123, 0. , 0. ], [0. , 0.80685455, 0. , 0. ], [0. , 0.76350731, 0. , 0. ], [0. , 0.76562130, 0. , 0. ], [0. , 0.87393230, 0. , 0. ], [0. , 0.82370549, 0. , 0. ], [0. , 0.89488727, 0. , 0. ], [0. , 0.84692216, 0. , 0. ], [0. , 0.80848366, 0. , 0. ], [0. , 0.80672985, 0. , 0. ], [0. , 0.75167274, 0. , 0. ], [0. , 0.74134314, 0. , 0. ], [0. , 0.73697317, 0. , 0. ], [0. , 0.81687951, 0. , 0. ], [0. , 0.61165327, 0. , 0. ], [0. , 0.69251174, 0. , 0. ], [0. , 0.70320952, 0. , 0. ], [0. , 0.86613262, 0. , 0. ], [0. , 0.83956414, 0. , 0. ], [0. , 0.60170650, 0. , 0. ], [0. , 0.69552523, 0. , 0. ], [0. , 0.77602792, 0. , 0. ], [0. , 0.68266684, 0. , 0. ], [0. , 0.84075797, 0. , 0. ], [0. , 0.79701507, 0. , 0. ]]), Tensor(shape=[2, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.21559088, 0. , 0. ], [0. , 0.21919683, 0. , 0. ]]), Tensor(shape=[34, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.69994599, 0. , 0. ], [0. , 0.67885721, 0. , 0. ], [0. , 0.66869640, 0. , 0. ], [0. , 0.58748865, 0. , 0. ], [0. , 0.51337993, 0. , 0. ], [0. , 0.73491317, 0. , 0. ], [0. , 0.72504926, 0. , 0. ], [0. , 0.82637447, 0. , 0. ], [0. , 0.76998556, 0. , 0. ], [0. , 0.82781368, 0. , 0. ], [0. , 0.89340150, 0. , 0. ], [0. , 0.80094981, 0. , 0. ], [0. , 0.82332069, 0. , 0. ], [0. , 0.81115031, 0. , 0. ], [0. , 0.74569678, 0. , 0. ], [0. , 0.80371106, 0. , 0. ], [0. , 0.86495912, 0. , 0. ], [0. , 0.85287029, 0. , 0. ], [0. , 0.89340341, 0. , 0. ], [0. , 0.78132838, 0. , 0. ], [0. , 0.90613228, 0. , 0. ], [0. , 0.64283836, 0. , 0. ], [0. , 0.61258817, 0. , 0. ], [0. , 0.59920675, 0. , 0. ], [0. , 0.59874809, 0. , 0. ], [0. , 0.68052024, 0. , 0. ], [0. , 0.68609858, 0. , 0. ], [0. , 0.86609662, 0. , 0. ], [0. , 0.60890704, 0. , 0. ], [0. , 0.86537188, 0. , 0. ], [0. , 0.88210642, 0. , 0. ], [0. , 0.68746102, 0. , 0. ], [0. , 0.73111236, 0. , 0. ], [0. , 0.83724433, 0. , 0. ]]), Tensor(shape=[106, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0.50688267, 0. , 0. , 0. ], [0.57206702, 0. , 0. , 0. ], [0.85267031, 0. , 0. , 0. ], [0.48482251, 0. , 0. , 0. ], [0.50186872, 0. , 0. , 0. ], [0.78307140, 0. , 0. , 0. ], [0.75822055, 0. , 0. , 0. ], [0.75034916, 0. , 0. , 0. ], [0.73755580, 0. , 0. , 0. ], [0.69863701, 0. , 0. , 0. ], [0.61544824, 0. , 0. , 0. ], [0.69417596, 0. , 0. , 0. ], [0.77172536, 0. , 0. , 0. ], [0.58661413, 0. , 0. , 0. ], [0.75315750, 0. , 0. , 0. ], [0.70431954, 0. , 0. , 0. ], [0.56309354, 0. , 0. , 0. ], [0.49433509, 0. , 0. , 0. ], [0.52818650, 0. , 0. , 0. ], [0.47487485, 0. , 0. , 0. ], [0.80439353, 0. , 0. , 0. ], [0.67314363, 0. , 0. , 0. ], [0.72916871, 0. , 0. , 0. ], [0.74127704, 0. , 0. , 0. ], [0.64851481, 0. , 0. , 0. ], [0.61427873, 0. , 0. , 0. ], [0.56599200, 0. , 0. , 0. ], [0.50206882, 0. , 0. , 0. ], [0.55693686, 0. , 0. , 0. ], [0.56577605, 0. , 0. , 0. ], [0.59108675, 0. , 0. , 0. ], [0.63443989, 0. , 0. , 0. ], [0.62632847, 0. , 0. , 0. ], [0.62411267, 0. , 0. , 0. ], [0.61616713, 0. , 0. , 0. ], [0.54493916, 0. , 0. , 0. ], [0.57163721, 0. , 0. , 0. ], [0.74053186, 0. , 0. , 0. ], [0.78581113, 0. , 0. , 0. ], [0.82777846, 0. , 0. , 0. ], [0.76590055, 0. , 0. , 0. ], [0.72276348, 0. , 0. , 0. ], [0.75337553, 0. , 0. , 0. ], [0.79108465, 0. , 0. , 0. ], [0.74514484, 0. , 0. , 0. ], [0.75225079, 0. , 0. , 0. ], [0.66042727, 0. , 0. , 0. ], [0.87956059, 0. , 0. , 0. ], [0.67525434, 0. , 0. , 0. ], [0.76588297, 0. , 0. , 0. ], [0.78135502, 0. , 0. , 0. ], [0.54897016, 0. , 0. , 0. ], [0.50960451, 0. , 0. , 0. ], [0.50124788, 0. , 0. , 0. ], [0.49484786, 0. , 0. , 0. ], [0.77984524, 0. , 0. , 0. ], [0.81153399, 0. , 0. , 0. ], [0.87664282, 0. , 0. , 0. ], [0.45983738, 0. , 0. , 0. ], [0.79543668, 0. , 0. , 0. ], [0.87540305, 0. , 0. , 0. ], [0.86274230, 0. , 0. , 0. ], [0.70562279, 0. , 0. , 0. ], [0.71385777, 0. , 0. , 0. ], [0.75141561, 0. , 0. , 0. ], [0.82321399, 0. , 0. , 0. ], [0.68214154, 0. , 0. , 0. ], [0.67414260, 0. , 0. , 0. ], [0.75977373, 0. , 0. , 0. ], [0.77518922, 0. , 0. , 0. ], [0.80970544, 0. , 0. , 0. ], [0.82668239, 0. , 0. , 0. ], [0.84573668, 0. , 0. , 0. ], [0.48632494, 0. , 0. , 0. ], [0.49824855, 0. , 0. , 0. ], [0.82972980, 0. , 0. , 0. ], [0.87980616, 0. , 0. , 0. ], [0.90955997, 0. , 0. , 0. ], [0.79996544, 0. , 0. , 0. ], [0.46049288, 0. , 0. , 0. ], [0.48514697, 0. , 0. , 0. ], [0.76897264, 0. , 0. , 0. ], [0.77370518, 0. , 0. , 0. ], [0.81103837, 0. , 0. , 0. ], [0.80775112, 0. , 0. , 0. ], [0.85463095, 0. , 0. , 0. ], [0.71391958, 0. , 0. , 0. ], [0.75137877, 0. , 0. , 0. ], [0.74476767, 0. , 0. , 0. ], [0.79629129, 0. , 0. , 0. ], [0.91742295, 0. , 0. , 0. ], [0.80613947, 0. , 0. , 0. ], [0.84692115, 0. , 0. , 0. ], [0.86090416, 0. , 0. , 0. ], [0.86716104, 0. , 0. , 0. ], [0.85691297, 0. , 0. , 0. ], [0.93505096, 0. , 0. , 0. ], [0.91276294, 0. , 0. , 0. ], [0.63260025, 0. , 0. , 0. ], [0.64534372, 0. , 0. , 0. ], [0.62422609, 0. , 0. , 0. ], [0.66048372, 0. , 0. , 0. ], [0.75661224, 0. , 0. , 0. ], [0.59765023, 0. , 0. , 0. ], [0.57793820, 0. , 0. , 0. ], [0.58668137, 0. , 0. , 0. ]]), Tensor(shape=[162, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0.85384524, 0. , 0. , 0. ], [0.77138716, 0. , 0. , 0. ], [0.81015754, 0. , 0. , 0. ], [0.82441390, 0. , 0. , 0. ], [0.82595021, 0. , 0. , 0. ], [0.66928220, 0. , 0. , 0. ], [0.76626748, 0. , 0. , 0. ], [0.71881771, 0. , 0. , 0. ], [0.82235283, 0. , 0. , 0. ], [0.83707732, 0. , 0. , 0. ], [0.84330469, 0. , 0. , 0. ], [0.89843857, 0. , 0. , 0. ], [0.80523157, 0. , 0. , 0. ], [0.79653782, 0. , 0. , 0. ], [0.78994846, 0. , 0. , 0. ], [0.61709237, 0. , 0. , 0. ], [0.66235536, 0. , 0. , 0. ], [0.59330541, 0. , 0. , 0. ], [0.65395623, 0. , 0. , 0. ], [0.71792835, 0. , 0. , 0. ], [0.69222194, 0. , 0. , 0. ], [0.57209796, 0. , 0. , 0. ], [0.79988295, 0. , 0. , 0. ], [0.66293257, 0. , 0. , 0. ], [0.61103839, 0. , 0. , 0. ], [0.53627390, 0. , 0. , 0. ], [0.82854778, 0. , 0. , 0. ], [0.61005104, 0. , 0. , 0. ], [0.56550223, 0. , 0. , 0. ], [0.84754205, 0. , 0. , 0. ], [0.70446533, 0. , 0. , 0. ], [0.69730854, 0. , 0. , 0. ], [0.66102898, 0. , 0. , 0. ], [0.70598447, 0. , 0. , 0. ], [0.73950773, 0. , 0. , 0. ], [0.72018379, 0. , 0. , 0. ], [0.76465851, 0. , 0. , 0. ], [0.79277009, 0. , 0. , 0. ], [0.73339736, 0. , 0. , 0. ], [0.75805283, 0. , 0. , 0. ], [0.62588561, 0. , 0. , 0. ], [0.79114997, 0. , 0. , 0. ], [0.69885045, 0. , 0. , 0. ], [0.67506826, 0. , 0. , 0. ], [0.64115620, 0. , 0. , 0. ], [0.59224486, 0. , 0. , 0. ], [0.67857182, 0. , 0. , 0. ], [0.66637903, 0. , 0. , 0. ], [0.70805985, 0. , 0. , 0. ], [0.66790730, 0. , 0. , 0. ], [0.70383751, 0. , 0. , 0. ], [0.72604245, 0. , 0. , 0. ], [0.77638966, 0. , 0. , 0. ], [0.78002626, 0. , 0. , 0. ], [0.60401541, 0. , 0. , 0. ], [0.73851293, 0. , 0. , 0. ], [0.61813200, 0. , 0. , 0. ], [0.67572314, 0. , 0. , 0. ], [0.61393011, 0. , 0. , 0. ], [0.82057035, 0. , 0. , 0. ], [0.72230297, 0. , 0. , 0. ], [0.64533126, 0. , 0. , 0. ], [0.65301186, 0. , 0. , 0. ], [0.71598887, 0. , 0. , 0. ], [0.65128034, 0. , 0. , 0. ], [0.66564101, 0. , 0. , 0. ], [0.81376451, 0. , 0. , 0. ], [0.70209551, 0. , 0. , 0. ], [0.67204469, 0. , 0. , 0. ], [0.61703891, 0. , 0. , 0. ], [0.67454875, 0. , 0. , 0. ], [0.55605692, 0. , 0. , 0. ], [0.51818162, 0. , 0. , 0. ], [0.54316092, 0. , 0. , 0. ], [0.74889916, 0. , 0. , 0. ], [0.59550554, 0. , 0. , 0. ], [0.59417033, 0. , 0. , 0. ], [0.66260523, 0. , 0. , 0. ], [0.62353241, 0. , 0. , 0. ], [0.58553427, 0. , 0. , 0. ], [0.72344279, 0. , 0. , 0. ], [0.58304411, 0. , 0. , 0. ], [0.56107742, 0. , 0. , 0. ], [0.63301420, 0. , 0. , 0. ], [0.58285689, 0. , 0. , 0. ], [0.54628366, 0. , 0. , 0. ], [0.60039490, 0. , 0. , 0. ], [0.64655310, 0. , 0. , 0. ], [0.57376450, 0. , 0. , 0. ], [0.64689916, 0. , 0. , 0. ], [0.60390919, 0. , 0. , 0. ], [0.58082515, 0. , 0. , 0. ], [0.81105965, 0. , 0. , 0. ], [0.65494865, 0. , 0. , 0. ], [0.61262441, 0. , 0. , 0. ], [0.64531785, 0. , 0. , 0. ], [0.54732680, 0. , 0. , 0. ], [0.75222623, 0. , 0. , 0. ], [0.60646009, 0. , 0. , 0. ], [0.64986861, 0. , 0. , 0. ], [0.64232862, 0. , 0. , 0. ], [0.59608883, 0. , 0. , 0. ], [0.58442408, 0. , 0. , 0. ], [0.80866414, 0. , 0. , 0. ], [0.78395784, 0. , 0. , 0. ], [0.76795959, 0. , 0. , 0. ], [0.76506197, 0. , 0. , 0. ], [0.89510280, 0. , 0. , 0. ], [0.70077795, 0. , 0. , 0. ], [0.74770880, 0. , 0. , 0. ], [0.84072989, 0. , 0. , 0. ], [0.82138550, 0. , 0. , 0. ], [0.90455621, 0. , 0. , 0. ], [0.78854728, 0. , 0. , 0. ], [0.82143730, 0. , 0. , 0. ], [0.79440886, 0. , 0. , 0. ], [0.74615198, 0. , 0. , 0. ], [0.77190030, 0. , 0. , 0. ], [0.82681686, 0. , 0. , 0. ], [0.78723699, 0. , 0. , 0. ], [0.80873525, 0. , 0. , 0. ], [0.83907533, 0. , 0. , 0. ], [0.77519995, 0. , 0. , 0. ], [0.80858010, 0. , 0. , 0. ], [0.81792694, 0. , 0. , 0. ], [0.76252681, 0. , 0. , 0. ], [0.67748964, 0. , 0. , 0. ], [0.71817672, 0. , 0. , 0. ], [0.86118239, 0. , 0. , 0. ], [0.79983723, 0. , 0. , 0. ], [0.68052560, 0. , 0. , 0. ], [0.65836698, 0. , 0. , 0. ], [0.64469498, 0. , 0. , 0. ], [0.78980172, 0. , 0. , 0. ], [0.75080645, 0. , 0. , 0. ], [0.71863210, 0. , 0. , 0. ], [0.82037532, 0. , 0. , 0. ], [0.80135566, 0. , 0. , 0. ], [0.67983502, 0. , 0. , 0. ], [0.69209093, 0. , 0. , 0. ], [0.74082476, 0. , 0. , 0. ], [0.72651953, 0. , 0. , 0. ], [0.70620745, 0. , 0. , 0. ], [0.69917369, 0. , 0. , 0. ], [0.68623567, 0. , 0. , 0. ], [0.66603392, 0. , 0. , 0. ], [0.62667650, 0. , 0. , 0. ], [0.64935309, 0. , 0. , 0. ], [0.61609161, 0. , 0. , 0. ], [0.61403608, 0. , 0. , 0. ], [0.65267801, 0. , 0. , 0. ], [0.66597158, 0. , 0. , 0. ], [0.66530842, 0. , 0. , 0. ], [0.60904258, 0. , 0. , 0. ], [0.73026133, 0. , 0. , 0. ], [0.78675425, 0. , 0. , 0. ], [0.84288836, 0. , 0. , 0. ], [0.75391120, 0. , 0. , 0. ], [0.73508364, 0. , 0. , 0. ], [0.74715060, 0. , 0. , 0. ], [0.64120972, 0. , 0. , 0. ], [0.65105647, 0. , 0. , 0. ]]), Tensor(shape=[27, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0.76053971, 0. , 0. ], [0. , 0.68215823, 0. , 0. ], [0. , 0.72128630, 0. , 0. ], [0. , 0.68419820, 0. , 0. ], [0. , 0.70152640, 0. , 0. ], [0. , 0.72409433, 0. , 0. ], [0. , 0.65370315, 0. , 0. ], [0. , 0.71758753, 0. , 0. ], [0. , 0.67804456, 0. , 0. ], [0. , 0.68409306, 0. , 0. ], [0. , 0.71926004, 0. , 0. ], [0. , 0.68731147, 0. , 0. ], [0. , 0.71184224, 0. , 0. ], [0. , 0.77135265, 0. , 0. ], [0. , 0.77010667, 0. , 0. ], [0. , 0.73166639, 0. , 0. ], [0. , 0.77004123, 0. , 0. ], [0. , 0.73480952, 0. , 0. ], [0. , 0.72676665, 0. , 0. ], [0. , 0.78714591, 0. , 0. ], [0. , 0.78629529, 0. , 0. ], [0. , 0.76406574, 0. , 0. ], [0. , 0.73293906, 0. , 0. ], [0. , 0.74265724, 0. , 0. ], [0. , 0.80331630, 0. , 0. ], [0. , 0.86788207, 0. , 0. ], [0. , 0.73268038, 0. , 0. ]]), Tensor(shape=[6, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True, [[0. , 0. , 0. , 0.66207230], [0. , 0. , 0. , 0.59610426], [0. , 0. , 0. , 0.77381659], [0. , 0. , 0. , 0.70487583], [0. , 0. , 0. , 0.78613091], [0. , 0. , 0. , 0.76987886]])]

[paddle-bot-old]

您好，我们已经收到了您的问题，会安排技术人员尽快解答您的问题，请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时，您也可以通过查看官网API文档、常见问题、历史Issue、AI社区来寻求解答。祝您生活愉快～

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the API，FAQ，Github Issue and AI community to get the answer.Have a nice day!

[Wei-JL]

如果可以，我可以把BML上的代码公开。 执行train文件就行。会训练到中途报错。
本机运行没问题，在本机上环境是CUDA:11.3 CUDNN:8.2.1 paddle2.2.2。

[Wei-JL]

我仔细的观察了列表reg_targets中的所有tensor，shape的第二个维度都是4, 因为其中出现shape[0, 4]的tensor，我用randn生成tensor的shape[0, 4] 加入到列表中 测试了也是可以直接concat的。 就是不知道为什么会在中间报维度的错误，求解答~~~ 大佬

[Wei-JL]

然后我复现了一下，
`import paddle
import numpy as np

_list = []
x1 = paddle.randn([1, 4])
x2 = paddle.randn([32, 4])
x3 = paddle.randn([12, 4])

_list.append(x2)
_list.append(x1)
_list.append(x3)
out2 = paddle.concat(_list, axis=0)
print(out2)`

输出是正确的

[zhangting2020]

你的意思是你换了运行环境吗？检查下paddle版本一样吗

[Wei-JL]

你的意思是你换了运行环境吗？检查下paddle版本一样吗

我再本地运行没事的， 我放到ai studio 上 运行一半报上面的错误

[Wei-JL]

我也不知道是不是版本原因 可能是本地运气好 没遇见shape=[1, 4]的 我本地判断一下有没有[1, 4]的

[Wei-JL]

本地有[1, 4] 也不报错。 是tensor的问题吗？ 把tensor shape=[1, 4]在concat里自动变成了[4]？ 怎么避免这个问题呢
我再里面加判断吗

[Wei-JL]

Epoch:36/100
Total Loss: 9.121 || Val Loss: 5.917 
Start Train
Epoch 37/100:  14%|▉      | 4/29 [00:07<00:48,  1.96s/it, loss=6.42, lr=0.00094]
报错tensor为 ： Tensor(shape=[1, 4], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
       [[0.        , 0.        , 0.21279770, 0.        ]]) ==========tensor的shape为 ： [1, 4]

Traceback (most recent call last):
  File "train.py", line 252, in <module>
    epoch_step, epoch_step_val, gen, gen_val, end_epoch, Cuda)
  File "/home/aistudio/yolox_paddle/utils/utils_fit.py", line 39, in fit_one_epoch
    loss_value = yolo_loss(outputs, targets)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 917, in __call__
    return self._dygraph_call_func(*inputs, **kwargs)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 907, in _dygraph_call_func
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/yolox_paddle/nets/yolo_training.py", line 146, in forward
    return self.get_losses(x_shifts, y_shifts, expanded_strides, labels, paddle.concat(outputs, 1))
  File "/home/aistudio/yolox_paddle/nets/yolo_training.py", line 252, in get_losses
    reg_targets, reg_targets_bool = concat_axis(reg_targets)
  File "/home/aistudio/yolox_paddle/nets/yolo_training.py", line 55, in concat_axis
    ten_tem = paddle.concat([ten_tem, ten], axis=0)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/tensor/manipulation.py", line 345, in concat
    return paddle.fluid.layers.concat(input=x, axis=axis, name=name)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/layers/tensor.py", line 327, in concat
    return _C_ops.concat(input, 'axis', axis)
ValueError: (InvalidArgument) The shape of input[0] and input[1] is expected to be equal.But received input[0]'s shape = [112, 4], input[1]'s shape = [4].
  [Hint: Expected inputs_dims[i].size() == out_dims.size(), but received inputs_dims[i].size():1 != out_dims.size():2.] (at /paddle/paddle/fluid/operators/concat_op.h:40)
  [operator < concat > error]

第5行日志可以看见shape=[1, 4] 调用paddle.concat后自动变成shape = [4] 导致整个程序报错:
ValueError: (InvalidArgument) The shape of input[0] and input[1] is expected to be equal.But received input[0]'s shape = [112, 4], input[1]'s shape = [4].

[Wei-JL]

你的意思是你换了运行环境吗？检查下paddle版本一样吗

大佬，有点急~ 您有空的话能给个回复吗？

[zhangting2020]

你的意思是你换了运行环境吗？检查下paddle版本一样吗

大佬，有点急~ 您有空的话能给个回复吗？

我测试你的case确实不会报错，需要首先确认下版本。已经联系了相关同学，但是因为已经是下班时间，可能需要等等

[Wei-JL]

你的意思是你换了运行环境吗？检查下paddle版本一样吗

大佬，有点急~ 您有空的话能给个回复吗？

我测试你的case确实不会报错，需要首先确认下版本。已经联系了相关同学，但是因为已经是下班时间，可能需要等等

用小例子测试确实不会有问题，比如:

实在不行我可以把我的BML开源，您可以fork运行一下。 我前面的第二个日志输出了，确实出现了这样的情况，运行的是.py文件

[Wei-JL]

项目“yolox”共享链接(有效期三天)：https://aistudio.baidu.com/studio/project/partial/verify/3801194/deaa3918a1cc405ca92bdd14d7038f94

直接从上到下运行就可以了，直到!python train.py

我崽本地训练了15轮都没有出问题 整个数据遍历了15遍的

[Wei-JL]

里面可能有import torch的忘了删 但是没有使用

[Wei-JL]

项目“yolox”共享链接(有效期三天)：https://aistudio.baidu.com/studio/project/partial/verify/3801194/5626331e3865408ba7852a1aa728690e