Skip to content

Commit

Permalink
Update scripts/i_jepa_mask_region_simulation.py
Browse files Browse the repository at this point in the history
Co-authored-by: zassou <[email protected]>
  • Loading branch information
Geson-anko and zassou65535 authored Oct 22, 2024
1 parent a14ee00 commit 272c558
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/i_jepa_mask_region_simulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@


def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Simulate mask region for BoolIJEPAMultiBlockMaskCollator")
parser = argparse.ArgumentParser(description="Calculate the ratio of the masked regions output from BoolIJEPAMultiBlockMaskCollator by simulation.")
parser.add_argument("--image_size", type=int, nargs=2, default=[144, 144], help="Image size (height, width)")
parser.add_argument("--patch_size", type=int, nargs=2, default=[12, 12], help="Patch size (height, width)")
parser.add_argument("--mask_scale", type=float, nargs=2, default=[0.10, 0.25], help="Mask scale range")
Expand Down

0 comments on commit 272c558

Please sign in to comment.