Spaces:
Sleeping
Sleeping
Search commited on
Commit ·
607c4cf
1
Parent(s): 8769d1f
auto: sync run_qwen_anchor_geometry_profile_sweep.py
Browse files
scripts/run_qwen_anchor_geometry_profile_sweep.py
CHANGED
|
@@ -245,29 +245,39 @@ def build_markdown_report(
|
|
| 245 |
return "\n".join(lines)
|
| 246 |
|
| 247 |
|
| 248 |
-
def
|
| 249 |
parser = argparse.ArgumentParser(description="Sweep anchor span profiles for geometry-only crystallization analysis.")
|
| 250 |
-
parser.add_argument("--model", type=str, default="Qwen/Qwen3.5-4B")
|
| 251 |
parser.add_argument("--device", type=str, default="cpu")
|
| 252 |
-
parser.add_argument("--max_length", type=int, default=160)
|
| 253 |
parser.add_argument(
|
| 254 |
"--profiles",
|
|
|
|
|
|
|
| 255 |
nargs="+",
|
| 256 |
default=list(list_anchor_span_profiles()),
|
| 257 |
)
|
| 258 |
-
parser.add_argument("--case_name", type=str, default=None)
|
| 259 |
parser.add_argument("--limit", type=int, default=None)
|
| 260 |
parser.add_argument(
|
| 261 |
"--output_json",
|
|
|
|
|
|
|
| 262 |
type=Path,
|
| 263 |
default=ROOT / "archive" / "qwen_anchor_geometry_profile_sweep.json",
|
| 264 |
)
|
| 265 |
parser.add_argument(
|
| 266 |
"--output_md",
|
|
|
|
|
|
|
| 267 |
type=Path,
|
| 268 |
default=ROOT / "docs" / "research" / "qwen_anchor_geometry_profile_sweep.md",
|
| 269 |
)
|
| 270 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 271 |
|
| 272 |
for profile in args.profiles:
|
| 273 |
if profile not in list_anchor_span_profiles():
|
|
|
|
| 245 |
return "\n".join(lines)
|
| 246 |
|
| 247 |
|
| 248 |
+
def parse_args(argv: list[str] | None = None) -> argparse.Namespace:
|
| 249 |
parser = argparse.ArgumentParser(description="Sweep anchor span profiles for geometry-only crystallization analysis.")
|
| 250 |
+
parser.add_argument("--model", "--model-name", dest="model", type=str, default="Qwen/Qwen3.5-4B")
|
| 251 |
parser.add_argument("--device", type=str, default="cpu")
|
| 252 |
+
parser.add_argument("--max_length", "--max-length", dest="max_length", type=int, default=160)
|
| 253 |
parser.add_argument(
|
| 254 |
"--profiles",
|
| 255 |
+
"--anchor-profile",
|
| 256 |
+
dest="profiles",
|
| 257 |
nargs="+",
|
| 258 |
default=list(list_anchor_span_profiles()),
|
| 259 |
)
|
| 260 |
+
parser.add_argument("--case_name", "--case-name", dest="case_name", type=str, default=None)
|
| 261 |
parser.add_argument("--limit", type=int, default=None)
|
| 262 |
parser.add_argument(
|
| 263 |
"--output_json",
|
| 264 |
+
"--output-json",
|
| 265 |
+
dest="output_json",
|
| 266 |
type=Path,
|
| 267 |
default=ROOT / "archive" / "qwen_anchor_geometry_profile_sweep.json",
|
| 268 |
)
|
| 269 |
parser.add_argument(
|
| 270 |
"--output_md",
|
| 271 |
+
"--output-md",
|
| 272 |
+
dest="output_md",
|
| 273 |
type=Path,
|
| 274 |
default=ROOT / "docs" / "research" / "qwen_anchor_geometry_profile_sweep.md",
|
| 275 |
)
|
| 276 |
+
return parser.parse_args(argv)
|
| 277 |
+
|
| 278 |
+
|
| 279 |
+
def main() -> None:
|
| 280 |
+
args = parse_args()
|
| 281 |
|
| 282 |
for profile in args.profiles:
|
| 283 |
if profile not in list_anchor_span_profiles():
|