diff --git a/requirements-colab.txt b/requirements-colab.txt index 1d0f166..9d638f5 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -9,3 +9,5 @@ scipy==1.4.1 colorama tensorflow-gpu==2.4.0 tf2onnx==1.9.3 +tensorboardX +crc32c diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 41bcc29..1cc4902 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -12,3 +12,5 @@ pyqt5 tf2onnx==1.9.3 Flask==1.1.1 flask-socketio==4.2.1 +tensorboardX +crc32c