Adds note which version of TFC is required for each model.

PiperOrigin-RevId: 349240922
Change-Id: I9551442b5ec5f8225896bbecffb646431c37a512

Author: Johannes Ballé

models/bls2017.py

@@ -26,6 +26,8 @@
 own experiments. To reproduce the exact results from the paper, tuning of hyper-
 parameters may be necessary. To compress images with published models, see
 `tfci.py`.
+
+Currently, this script requires tensorflow-compression v1.3.
 """
 
 import argparse

models/bmshj2018.py

@@ -24,6 +24,8 @@
 own experiments. To reproduce the exact results from the paper, tuning of hyper-
 parameters may be necessary. To compress images with published models, see
 `tfci.py`.
+
+Currently, this script requires tensorflow-compression v1.3.
 """
 
 import argparse

models/ms2020.py

@@ -24,6 +24,8 @@
 own experiments. To reproduce the exact results from the paper, tuning of hyper-
 parameters may be necessary. To compress images with published models, see
 `tfci.py`.
+
+This script requires tensorflow-compression v2.x.
 """
 
 import argparse

models/tfci.py

@@ -16,6 +16,8 @@
 
 Use this script to compress images with pre-trained models as published. See the
 'models' subcommand for a list of available models.
+
+Currently, this script requires tensorflow-compression v1.3.
 """
 
 import argparse

models/toy_sources/toy_sources.ipynb

@@ -20,7 +20,9 @@
         "\n",
         "\u003e \"Neural Networks Optimally Compress the Sawbridge\"\u003cbr /\u003e\n",
         "\u003e A. B. Wagner, J. Ballé\u003cbr /\u003e\n",
-        "\u003e https://arxiv.org/abs/2011.05065\n"
+        "\u003e https://arxiv.org/abs/2011.05065\n",
+        "\n",
+        "This code requires tensorflow-compression v2.x.\n"
       ]
     },
     {