InvalidArgumentError Traceback (most recent call last)
<ipython-input-11-5600f6efe627> in <module>()
 5 callbacks=callbacks,
 6 validation_data=valid_dataloader,
----> 7 validation_steps=len(valid_dataloader)
 8 #initial_epoch = 41
 9 )
8 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/eager/execute.py in quick_execute(op_name, num_outputs, inputs, attrs, ctx, name)
 58 ctx.ensure_initialized()
 59 tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,
---> 60 inputs, attrs, num_outputs)
 61 except core._NotOkStatusException as e:
 62 if name is not None:
InvalidArgumentError: All dimensions except 3 must match. Input 1 has shape [6 48 48 512] and doesn't match input 0 with shape [6 32 32 320].
	 [[node gradient_tape/functional_1/psp_concat/ConcatOffset (defined at <ipython-input-11-5600f6efe627>:7) ]] [Op:__inference_train_function_30002]
Function call stack:
train_function