кстати, на счет смены нормализации, разве batchnorm который делает нормализацию относительно измерения channels с batch_size=1 это не одно и то же что instancenorm ?
along (H,W) for each channel (InstanceNorm) или Because the Batch Normalization is done over the C dimension, computing statistics on (N, H, W) slices, it’s common terminology to call this Spatial Batch Normalization. (BatchNorm Pytorch)