diff --git a/avg_checkpoints.py b/avg_checkpoints.py index a6921224..1f7604b0 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 94f184d1..a8edcc91 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 88406997..1985669e 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 77be3602..89efb1fb 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 f0fcd2af..1dd5f39e 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 8ad9cb1f..6e8eff7a 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