Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Efficientnet lite #141

Merged
merged 7 commits into from
Jul 31, 2020
Merged

Efficientnet lite #141

merged 7 commits into from
Jul 31, 2020

Conversation

slahiruk
Copy link
Contributor

Efficientnet-lite related commits for ck-tensorflow

# SPDX-License-Identifier: BSD-3-Clause.
# See CK LICENSE.txt for licensing details.
#
# MobileNet for TensorFlow
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

MobileNet -> EfficientNet


echo "Keeping file ${PACKAGE_NAME_EFFICIENTNET_LITE_WITH_PRECISION}.tflite"

if [ "${MODEL_EFFICIENTNET_LITE_PRECISION}" == "fp32" ]; then
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please move common code out of the branches:

  if [ "${MODEL_EFFICIENTNET_LITE_PRECISION}" == "fp32" ]; then
    file_to_remove="${PACKAGE_NAME_EFFICIENTNET_LITE}-int8.tflite"
  else
    file_to_remove="${PACKAGE_NAME_EFFICIENTNET_LITE}-fp32.tflite"
  fi
  echo "Removing file ${file_to_remove}"
  rm ${file_to_remove}

echo "Removing file ${file_to_remove}"
rm ${file_to_remove}
fi

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please remove empty lines.

echo
echo "Remove temporary files ..."
rm ${PACKAGE_NAME}

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please remove everything below this line.

tar -zxvf ${PACKAGE_NAME}


#Efficientnet-lite unpacks into subdirectory
Copy link
Contributor

@psyhtest psyhtest Jul 31, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

# EfficientNet-Lite



#Efficientnet-lite unpacks into subdirectory
if [[ -d ${PACKAGE_NAME_EFFICIENTNET_LITE} ]]; then
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Don't strictly need this if as only dealing with EfficientNet in this script.

@ens-lg4 ens-lg4 merged commit bbd1c07 into ctuning:master Jul 31, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants