Add PyTorch to ONNX section to training notebooks (#903)

- Add PyTorch to ONNX section to the following training notebooks: - Phishing - Root Cause - Log Parsing - SID - Update Phishing and SID models to work with torch 1.13 Closes #892 Closes #529 Authors: - Eli Fajardo (https://github.com/efajardo-nv) - Michael Demoret (https://github.com/mdemoret-nv) Approvers: - Michael Demoret (https://github.com/mdemoret-nv) URL: #903
nv-morpheus · May 1, 2023 · a713d95 · a713d95
1 parent 070897d
commit a713d95
Show file tree

Hide file tree

Showing 15 changed files with 447 additions and 196 deletions.
diff --git a/models/phishing-models/phishing-bert-20221115.onnx b/models/phishing-models/phishing-bert-20221115.onnx
diff --git a/models/phishing-models/phishing-bert-20221115.pt b/models/phishing-models/phishing-bert-20221115.pt
diff --git a/models/phishing-models/phishing-bert-20230421.onnx b/models/phishing-models/phishing-bert-20230421.onnx
diff --git a/models/phishing-models/phishing-bert-20230421.pt b/models/phishing-models/phishing-bert-20230421.pt
diff --git a/models/sid-models/sid-minibert-20211021.onnx b/models/sid-models/sid-minibert-20211021.onnx
diff --git a/models/sid-models/sid-minibert-20211021.pth b/models/sid-models/sid-minibert-20211021.pth
diff --git a/models/sid-models/sid-minibert-20230424.onnx b/models/sid-models/sid-minibert-20230424.onnx
diff --git a/models/sid-models/sid-minibert-20230424.pt b/models/sid-models/sid-minibert-20230424.pt