Merge pull request #418 from kecsap/master

Change the Python interpreter to Python 3.x in the scripts
pull/421/head
Ross Wightman 4 years ago committed by GitHub
commit db6128db0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

@ -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.

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

@ -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.

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

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

Loading…
Cancel
Save