Skip to content

Commit 09a34e6

Browse files
Johannes Ball?copybara-github
authored andcommitted
Updates docs to refer to TF 1.15.
Also enables explicit convolution mode by default, since it is now supported. PiperOrigin-RevId: 281167413 Change-Id: I96bccf31fa405db23fe8c06cfd6543da7998be72
1 parent e750212 commit 09a34e6

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

README.md

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -50,15 +50,17 @@ packages using the `pip` command. Refer to the
5050
[TensorFlow installation instructions](https://www.tensorflow.org/install/pip)
5151
for more information on how to set up such a Python environment.
5252

53-
The current version of tensorflow-compression requires TensorFlow 1.14 or
54-
above. You can install TensorFlow from any source. To install it via `pip`, run
55-
the following command:
53+
The current version of tensorflow-compression requires TensorFlow 1.15. For
54+
TensorFlow 1.14 or earlier, see our
55+
[previous releases](https://github.com/tensorflow/compression/releases). You can
56+
install TensorFlow from any source. To install it via `pip`, run the following
57+
command:
5658
```bash
57-
pip install tensorflow-gpu
59+
pip install tensorflow-gpu==1.15
5860
```
5961
for GPU support, or
6062
```bash
61-
pip install tensorflow
63+
pip install tensorflow==1.15
6264
```
6365
for CPU-only.
6466

@@ -72,7 +74,7 @@ To test that the installation works correctly, you can run the unit tests with
7274
```bash
7375
python -m tensorflow_compression.python.all_test
7476
```
75-
Once the command finishes, you should see a message ```OK (skipped=11)``` or
77+
Once the command finishes, you should see a message ```OK (skipped=12)``` or
7678
similar in the last line.
7779

7880
### Docker
@@ -102,7 +104,7 @@ and tensorflow-compression with GPU support:
102104
```bash
103105
conda create --name ENV_NAME python=3.6 cudatoolkit=10.0 cudnn
104106
conda activate ENV_NAME
105-
pip install tensorflow-gpu tensorflow-compression
107+
pip install tensorflow-gpu==1.15 tensorflow-compression
106108
```
107109

108110
## Usage

tensorflow_compression/python/layers/signal_conv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ def __init__(self, filters, kernel_support,
192192
corr=False, strides_down=1, strides_up=1, padding="valid",
193193
extra_pad_end=True, channel_separable=False,
194194
data_format="channels_last",
195-
activation=None, use_bias=False, use_explicit=False,
195+
activation=None, use_bias=False, use_explicit=True,
196196
kernel_initializer=tf.initializers.variance_scaling(),
197197
bias_initializer=tf.initializers.zeros(),
198198
kernel_regularizer=None, bias_regularizer=None,

0 commit comments

Comments
 (0)