File "d:\learn\Anaconda3\envs\py3.8\lib\site-packages\keras\engine\training.py", line 1160, in train_function *
return step_function(self, iterator)
File "d:\learn\Anaconda3\envs\py3.8\lib\site-packages\keras\engine\training.py", line 1146, in step_function **
outputs = model.distribute_strategy.run(run_step, args=(data,))
File "d:\learn\Anaconda3\envs\py3.8\lib\site-packages\keras\engine\training.py", line 1135, in run_step **
outputs = model.train_step(data)
File "d:\learn\Anaconda3\envs\py3.8\lib\site-packages\keras\engine\training.py", line 993, in train_step
...
ValueError: Input 0 of layer "TS-NAS" is incompatible with the layer: expected shape=(None, 256, 6), found shape=(128, 256, 29)
ValueError: in user code:
...