Untitled

if args.mask_type == 'tube':
            self.masked_position_generator = TubeMaskingGenerator(
                args.window_size, args.mask_ratio
            )
        elif args.mask_type == 'part_window':
            print(f"==> Note: use 'part_window' masking generator (window_size={args.part_win_size[1:]}, apply_symmetry={args.part_apply_symmetry})")
            self.masked_position_generator = TubeWindowMaskingGenerator(
                args.window_size, args.mask_ratio, win_size=args.part_win_size[1:], apply_symmetry=args.part_apply_symmetry
            )