Define the CutMix data augmentation function (tensorflow native 만 이용해서 custom augmentation 을 정의. tf.function 을 사용하는 code 코드 조각 참고)

@tf.function def cutmix(train_ds_one, train_ds_two): return image, label ... # Create the new dataset using our `cutmix` utility train_ds_cmu = ( train_ds.shuffle(1024) .map(cutmix, num_parallel_calls=AUTO) .batch(BATCH_SIZE) .prefetch(AUTO) )