diff --git a/avg_checkpoints.py b/avg_checkpoints.py index a69212248..1f7604b05 100755 --- a/avg_checkpoints.py +++ b/avg_checkpoints.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 """ Checkpoint Averaging Script This script averages all model weights for checkpoints in specified path that match diff --git a/clean_checkpoint.py b/clean_checkpoint.py index 94f184d1b..a8edcc915 100755 --- a/clean_checkpoint.py +++ b/clean_checkpoint.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 """ Checkpoint Cleaning Script Takes training checkpoints with GPU tensors, optimizer state, extra dict keys, etc. diff --git a/distributed_train.sh b/distributed_train.sh index 884069974..1985669e2 100755 --- a/distributed_train.sh +++ b/distributed_train.sh @@ -1,5 +1,5 @@ #!/bin/bash NUM_PROC=$1 shift -python -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@" +python3 -m torch.distributed.launch --nproc_per_node=$NUM_PROC train.py "$@" diff --git a/inference.py b/inference.py index 77be3602b..89efb1fb7 100755 --- a/inference.py +++ b/inference.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 """PyTorch Inference Script An example inference script that outputs top-k class ids for images in a folder into a csv. diff --git a/train.py b/train.py index f0fcd2af2..1dd5f39ed 100755 --- a/train.py +++ b/train.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 """ ImageNet Training Script This is intended to be a lean and easily modifiable ImageNet training script that reproduces ImageNet diff --git a/validate.py b/validate.py index 8ad9cb1f0..6e8eff7a3 100755 --- a/validate.py +++ b/validate.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 """ ImageNet Validation Script This is intended to be a lean and easily modifiable ImageNet validation script for evaluating pretrained