add MixConv2d() layer

2020-04-03 20:03:44 -07:00
parent eb9fb245aa
commit 00c1fdd805
4 changed files with 46 additions and 9 deletions
@@ -2,7 +2,7 @@
 FROM nvcr.io/nvidia/pytorch:20.03-py3

 # Install dependencies (pip or conda)
-RUN pip install -U gsutil
+RUN pip install -U gsutil thop
 # RUN pip install -U -r requirements.txt
 # RUN conda update -n base -c defaults conda
 # RUN conda install -y -c anaconda future numpy opencv matplotlib tqdm pillow