def get_batch_tfms(run_params):
    label_tfms = [IntToFloatTensor(div=((2 ** 16) - 1)), *aug_transforms(pad_mode=PadMode.Zeros, mult=1.0, do_flip=True, flip_vert=False, max_rotate=90.0, min_zoom=0.9, max_zoom=1.2, max_lighting=0.4, max_warp=0.4, p_affine=0.9, p_lighting=0.9, mode='bilinear', align_corners=True), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE'])]
    unlabel_tfms = [[IntToFloatTensor(div=((2 ** 16) - 1))]]
    if (run_params['SSL'] == run_params['SSL_FIX_MATCH']):
        weak_transform = [IntToFloatTensor(div=1), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE']), Flip()]
        unlabel_tfms.append(weak_transform)
        strong_transform = [IntToFloatTensor(div=1), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE']), Flip(), Rotate(180), Brightness(), Contrast(), RandomErasing()]
        unlabel_tfms.append(strong_transform)
    elif (run_params['SSL'] == run_params['SSL_MIX_MATCH']):
        unlabel_transform = [IntToFloatTensor(div=((2 ** 16) - 1)), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE']), Flip(), Rotate(180), Brightness(), Contrast()]
        unlabel_tfms.append(unlabel_transform)
    return (label_tfms, unlabel_tfms)
def get_batch_tfms(run_params):
    label_tfms = [IntToFloatTensor(div=((2 ** 16) - 1)), *aug_transforms(pad_mode=PadMode.Zeros, mult=1.0, do_flip=True, flip_vert=False, max_rotate=90.0, min_zoom=0.9, max_zoom=1.2, max_lighting=0.4, max_warp=0.4, p_affine=0.9, p_lighting=0.9, mode='bilinear', align_corners=True), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE'])]
    unlabel_tfms = [[IntToFloatTensor(div=((2 ** 16) - 1))]]
    if (run_params['SSL'] == run_params['SSL_FIX_MATCH']):
        weak_transform = [IntToFloatTensor(div=1), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE']), Flip()]
        unlabel_tfms.append(weak_transform)
        strong_transform = [IntToFloatTensor(div=1), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE']), Flip(), Rotate(180), Brightness(), Contrast(), RandomErasing()]
        unlabel_tfms.append(strong_transform)
    elif (run_params['SSL'] == run_params['SSL_MIX_MATCH']):
        unlabel_transform = [IntToFloatTensor(div=((2 ** 16) - 1)), RandomResizedCropGPU(run_params['RANDOM_RESIZE_CROP'], min_scale=run_params['RANDOM_MIN_SCALE']), Flip(), Rotate(180), Brightness(), Contrast()]
        unlabel_tfms.append(unlabel_transform)
    return (label_tfms, unlabel_tfms)