remove thop dependency

Dockerfile

@@ -2,7 +2,7 @@
 FROM nvcr.io/nvidia/pytorch:20.03-py3
 
 # Install dependencies (pip or conda)
-RUN pip install -U gsutil thop
+RUN pip install -U gsutil
 # RUN pip install -U -r requirements.txt
 
 # Create working directory