From d679425b351cd3145dbadbceef0bf6ed1a0bbd8b Mon Sep 17 00:00:00 2001 From: David Gardner <96306125+dagardner-nv@users.noreply.github.com> Date: Mon, 21 Oct 2024 12:38:50 -0700 Subject: [PATCH] Fix various linting and formatting errors (#89) * Fix various linting and formatting errors Authors: - David Gardner (https://github.com/dagardner-nv) Approvers: - https://github.com/shawn-davis URL: https://github.com/nv-morpheus/morpheus-experimental/pull/89 --- cyber-foundation/training-inference/prepare-prefix.py | 1 - cyber-foundation/training-inference/train-prefix.py | 2 -- 2 files changed, 3 deletions(-) diff --git a/cyber-foundation/training-inference/prepare-prefix.py b/cyber-foundation/training-inference/prepare-prefix.py index 7b12f37..e10bf59 100644 --- a/cyber-foundation/training-inference/prepare-prefix.py +++ b/cyber-foundation/training-inference/prepare-prefix.py @@ -39,7 +39,6 @@ # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE # SOFTWARE. -import json import os import pickle diff --git a/cyber-foundation/training-inference/train-prefix.py b/cyber-foundation/training-inference/train-prefix.py index 3506c49..425fcb5 100644 --- a/cyber-foundation/training-inference/train-prefix.py +++ b/cyber-foundation/training-inference/train-prefix.py @@ -53,8 +53,6 @@ from torch.distributed import init_process_group from torch.nn.parallel import DistributedDataParallel as DDP - - out_dir = '../model/' eval_interval = 100 log_interval = 1