Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions official/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,4 +28,6 @@ If you would like to make any fixes or improvements to the models, please [submi

The *Official Models* are made available as a Python module. To run the models and associated scripts, add the top-level ***/models*** folder to the Python path with the command: `export PYTHONPATH="$PYTHONPATH:/path/to/models"`

To install dependencies pass `-r official/requirements.txt` to pip. (i.e. `pip3 install --user -r official/requirements.txt`)

To make Official Models easier to use, we are planning to create a pip installable Official Models package. This is being tracked in [#917](https://github.com/tensorflow/models/issues/917).
2 changes: 2 additions & 0 deletions official/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
psutil>=5.4.3
py-cpuinfo>=3.3.0
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should tf-nightly be listed here too? I don't know if there's a way to do a conditional on gpu.

4 changes: 1 addition & 3 deletions official/utils/logging/logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,7 @@
"""Logging utilities for benchmark.
For collecting local environment metrics like CPU and memory, certain python
packages need be installed. Run the following commands for dependency packages:
> pip install --upgrade py-cpuinfo
> pip install --upgrade psutil
packages need be installed. See README for details.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No reason to make comments less explicit here.

"""
from __future__ import absolute_import
from __future__ import division
Expand Down