Skip to content

Commit

Permalink
Fix various linting and formatting errors (#89)
Browse files Browse the repository at this point in the history
* Fix various linting and formatting errors

Authors:
  - David Gardner (https://github.com/dagardner-nv)

Approvers:
  - https://github.com/shawn-davis

URL: #89
  • Loading branch information
dagardner-nv authored Oct 21, 2024
1 parent 8e0a1d2 commit d679425
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 3 deletions.
1 change: 0 additions & 1 deletion cyber-foundation/training-inference/prepare-prefix.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
2 changes: 0 additions & 2 deletions cyber-foundation/training-inference/train-prefix.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit d679425

Please sign in to comment.