ModerRAS
/

AniFileBERT

Token Classification

filename-parsing

Eval Results (legacy)

Model card Files Files and versions

ModerRAS commited on 7 days ago

Commit

c5ba3cb

·

1 Parent(s): 4df510c

colab: 加上onnx/onnxruntime依赖

Files changed (1) hide show

colab_train.py +1 -1

colab_train.py CHANGED Viewed

@@ -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")
 # ── 4. Verify GPU ──────────────────────────────────────────────
 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")
 # ── 4. Verify GPU ──────────────────────────────────────────────
 print("\n" + "=" * 60)