Lance_N:

You can convert data between channels-first and channels-last with a ‘permute dims’ tensor call. This can be done with a Lambda layer.


@Lance_N ,
Thank you for your reply.
I checked above link. I had already done everything I had to do.
Though I’ve already build channel-first model architecture, how do I have to use “Permute dims” ?
Thank you.