[EfficientNet/TF2] Initial release of EfficientNet V2

Author: Victor49152

TensorFlow2/Classification/ConvNets/.dockerignore

@@ -0,0 +1 @@
+output

TensorFlow2/Classification/ConvNets/efficientnet/.gitignore renamed to TensorFlow2/Classification/ConvNets/.gitignore

@@ -1,3 +1,13 @@
+config/efficientnet_v2/s_cfg_fumb.py
+config/efficientnet_v2/s_cfg_mb.py
+config/efficientnet_v1/b4_allfumb_cfg.py
+config/efficientnet_v1/b4_mixed_cfg.py
+debug.py
+output/
+src/
+.vscode/
+eval/
+log.json
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

TensorFlow2/Classification/ConvNets/efficientnet/Dockerfile renamed to TensorFlow2/Classification/ConvNets/Dockerfile

@@ -12,18 +12,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:21.02-tf2-py3
+ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:21.09-tf2-py3
 FROM ${FROM_IMAGE_NAME}
 
 RUN echo ${FROM_IMAGE_NAME} 
 
-LABEL Effnet_tf by subhankarg
+LABEL Effnetv2_tf2 by ntajbakhsh
 
-RUN rm -rf /workspace && mkdir -p /workspace
-ADD . /workspace
 WORKDIR /workspace
 COPY . .
 
-RUN pip --no-cache-dir --no-cache install --user -r requirements.txt
+RUN python -m pip install --upgrade pip && \
+    pip --no-cache-dir --no-cache install --user -r requirements.txt
 
-RUN pip install git+https://github.com/NVIDIA/dllogger
+ENV TF_XLA_FLAGS="--tf_xla_enable_lazy_compilation=false"
+RUN pip install git+https://github.com/NVIDIA/dllogger

TensorFlow2/Classification/ConvNets/efficientnet/LICENSE renamed to TensorFlow2/Classification/ConvNets/LICENSE

@@ -0,0 +1,3 @@
+This repository supports 
+  - EfficientNet v1-B0 and v1-B4. Please see [here](efficientnet_v1/README.md) for the complete guide.
+  - EfficientNet v2-S. Please see [here](efficientnet_v2/README.md) for the complete guide.