From 88762a2f8cc409fe15a9e6a049fe69ae3197fc49 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Tue, 2 Jun 2020 04:29:28 -0400 Subject: [PATCH] Specify PyTorch versions for examples (#4710) --- README.md | 2 +- examples/README.md | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 6beaf0025d..8dfe0c86ac 100644 --- a/README.md +++ b/README.md @@ -63,7 +63,7 @@ Choose the right framework for every part of a model's lifetime ## Installation -This repo is tested on Python 3.6+, PyTorch 1.0.0+ and TensorFlow 2.0. +This repo is tested on Python 3.6+, PyTorch 1.0.0+ (PyTorch 1.3.1+ for examples) and TensorFlow 2.0. You should install 🤗 Transformers in a [virtual environment](https://docs.python.org/3/library/venv.html). If you're unfamiliar with Python virtual environments, check out the [user guide](https://packaging.python.org/guides/installing-using-pip-and-virtual-environments/). diff --git a/examples/README.md b/examples/README.md index 372d73aa83..82a3610954 100644 --- a/examples/README.md +++ b/examples/README.md @@ -1,6 +1,7 @@ ## Examples Version 2.9 of `transformers` introduces a new [`Trainer`](https://github.com/huggingface/transformers/blob/master/src/transformers/trainer.py) class for PyTorch, and its equivalent [`TFTrainer`](https://github.com/huggingface/transformers/blob/master/src/transformers/trainer_tf.py) for TF 2. +Running the examples requires PyTorch 1.3.1+ or TensorFlow 2.0+. Here is the list of all our examples: - **grouped by task** (all official examples work for multiple models)