Skip to content

Commit

Permalink
docker: Drop caffe2 download progess bars (apache#5359)
Browse files Browse the repository at this point in the history
Change-Id: Ia15c3c8f41f75423814e559f6fdb062098f19464
  • Loading branch information
mshawcroft authored and dpankratz committed Apr 24, 2020
1 parent b1dec0c commit 7b52cf4
Showing 1 changed file with 17 additions and 3 deletions.
20 changes: 17 additions & 3 deletions docker/install/ubuntu_install_caffe2.sh
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,20 @@ set -e
set -u
set -o pipefail

python3 -m caffe2.python.models.download -i -f squeezenet
python3 -m caffe2.python.models.download -i -f resnet50
python3 -m caffe2.python.models.download -i -f vgg19
# caffe2.python.module.download generates a progress bar. in non
# interactive use this results in huge progress debris in the log
# files. There is no option to disable the progress bar so work
# around it by stripping the progress bar output

filter_progress_bar()
{
# Progress bars are the 'goto start of line' escape sequence
# ESC[1000D[ repeated, the end of the progress bar is the end of
# line. We can selectively remove progress bars by dropping lines
# that beging with the escape sequence.
sed "/^\x1b\[1000D/d"
}

python3 -m caffe2.python.models.download -i -f squeezenet | filter_progress_bar
python3 -m caffe2.python.models.download -i -f resnet50 | filter_progress_bar
python3 -m caffe2.python.models.download -i -f vgg19 | filter_progress_bar

0 comments on commit 7b52cf4

Please sign in to comment.