fluid.layers.split 对 batch size 这个通道进行split, 因为维度为-1,assert失败了
Created by: kebinC
File "thirdparty/paddlemodels/metric_learning/trainfast_aiflow_elem_multitask.py", line 360, in main
train_async(args)
File "thirdparty/paddlemodels/metric_learning/trainfast_aiflow_elem_multitask.py", line 228, in train_async
train_batch_split=train_batch_size)
File "thirdparty/paddlemodels/metric_learning/trainfast_aiflow_elem_multitask.py", line 183, in build_program
avg_cost, acc_top1, acc_top5, out = net_config(image, label, stn, model, args, dtype, is_train, train_batch_split)
File "thirdparty/paddlemodels/metric_learning/trainfast_aiflow_elem_multitask.py", line 124, in net_config
embedding_split = fluid.layers.split(out['embedding'], train_batch_split, dim=0)
File "/home/users/chenkaibing/env/paddle-release/python-gcc482-paddle/lib/python2.7/site-packages/paddle/fluid/layers/nn.py", line 6712, in split
dim], 'len(num_or_sections) must not be more than input.shape[dim].'
AssertionError: len(num_or_sections) must not be more than input.shape[dim].