Merge pull request #418 from kecsap/master

Change the Python interpreter to Python 3.x in the scripts

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

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.

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 "$@"
 

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.

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

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