colab: onnx导出改为非阻塞+补全onnxscript

colab_train.py

@@ -76,7 +76,7 @@ print("\n" + "=" * 60)
 print("STEP 3: Install dependencies")
 print("=" * 60)
 # Colab comes with PyTorch + CUDA pre-installed. Just install the extras.
-run("pip install transformers accelerate seqeval onnx onnxruntime")
+run("pip install transformers accelerate seqeval onnx onnxruntime onnxscript")
 
 # ── 4. Verify GPU ──────────────────────────────────────────────
 print("\n" + "=" * 60)
@@ -113,16 +113,19 @@ run(
     f"--no-shuffle"
 )
 
-# ── 7. Export ONNX ─────────────────────────────────────────────
+# ── 7. Export ONNX (optional) ──────────────────────────────────
 print("\n" + "=" * 60)
-print("STEP 7: Export ONNX (optional)")
+print("STEP 7: Export ONNX (optional — skip if it fails)")
 print("=" * 60)
 ONNX_OUT = os.path.join(SAVE_DIR, "..", "anime_filename_parser.onnx")
-run(
-    f"python export_onnx.py "
-    f"--model-dir {SAVE_DIR}/final "
-    f"--output {ONNX_OUT}"
-)
+try:
+    run(
+        f"python export_onnx.py "
+        f"--model-dir {SAVE_DIR}/final "
+        f"--output {ONNX_OUT}"
+    )
+except Exception as e:
+    print(f"[WARN] ONNX export skipped: {e}")
 
 # ── 8. Summary ─────────────────────────────────────────────────
 print("\n" + "=" * 60)