/run_torchbench

This repo contains a list of stand alone scripts for running torchbench models one by one.

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

run_torchbench

This repo contains a list of stand alone scripts for running torchbench models one by one.

Dependencies

1. Torch XLA2:

Please install following the instructions in https://github.com/pytorch/xla/tree/master/experimental/torch_xla2

2. TorchBench

Important: Please git clone torchbench in this directory and install from source

pip install torchvision torchaudio
git clone https://github.com/pytorch/benchmark.git
cd benchmark
python install.py # install model
# pip install . # install torchbench

NOTE: python install.py will install ALL models. If you are only interested in running one model, you can install it via python install.py model_name this way it will be much faster.

3. Run one torch bench model under torch xla2

python models/name_of_the_model.py

For example,

python models/BERT_pytorch.py

The scripts are identical per model except the model name. The goal is for people to add loggings and edit the script as they see fit. And able to focus just one model.