luping123 commited on
Commit
be717fd
·
1 Parent(s): 02581b3

card: CPU support

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -27,7 +27,9 @@ IntraSpanNet names what produced a flow from its first five packets: a security
27
 
28
  **DOI:** [10.5281/zenodo.23643146](https://zenodo.org/records/23643146)
29
 
30
- **Code and training guide:** [github.com/lupingQAQ/IntraSpanNet](https://github.com/lupingQAQ/IntraSpanNet) — [docs/TRAINING.md](https://github.com/lupingQAQ/IntraSpanNet/blob/main/docs/TRAINING.md)
 
 
31
 
32
  ## Model details
33
 
@@ -87,7 +89,7 @@ The test split is a flow-level half of the validation set, not a session-level h
87
 
88
  - `stunnel` has zero recall in the acceptance demo (0/30).
89
  - Flow-level test split; session-level generalization is not measured.
90
- - CPU inference is not supported. The GDN backbone uses Triton kernels and requires CUDA.
91
  - The model was trained on lab traffic. Production performance with different software versions, proxies, or NAT is unknown.
92
  - The malicious head is a secondary output and is not calibrated for production alerting.
93
 
 
27
 
28
  **DOI:** [10.5281/zenodo.23643146](https://zenodo.org/records/23643146)
29
 
30
+ **Code and training guide:** [github.com/lupingQAQ/IntraSpanNet](https://github.com/lupingQAQ/IntraSpanNet) — [docs/TRAINING.md](https://github.com/lupingQAQ/IntraSpanNet/blob/main/docs/TRAINING.md)
31
+
32
+ **Install:** `pip install intraspannet` runs on CPU; `pip install "intraspannet[gpu]"` adds the Triton kernel. Same weights either way.
33
 
34
  ## Model details
35
 
 
89
 
90
  - `stunnel` has zero recall in the acceptance demo (0/30).
91
  - Flow-level test split; session-level generalization is not measured.
92
+ - CPU inference uses a plain PyTorch port of the GDN block: correct, but about two seconds per flow per arm, so it is for inspection rather than a live probe. `pip install "intraspannet[gpu]"` enables the Triton path.
93
  - The model was trained on lab traffic. Production performance with different software versions, proxies, or NAT is unknown.
94
  - The malicious head is a secondary output and is not calibrated for production alerting.
95