Contents Menu Expand
AirPack 0.3.0 documentation
Logo
AirPack 0.3.0 documentation

Contents:

  • AirPack Overview
  • AirPack Installation
  • Training the Model
  • Perform Inference on AIR-T
  • Working with Data Sets
  • Next Steps
  • Frequently Asked Questions
  • AirPack Terms of Use
  • Documentation Archive
  • API Reference
    • airpack
      • airpack.deploy
        • airpack.deploy.onnx
        • airpack.deploy.trt
        • airpack.deploy.trt_utils
      • airpack.pytorch
        • airpack.pytorch.fileio
        • airpack.pytorch.model
      • airpack.tf1
        • airpack.tf1.fileio
        • airpack.tf1.model
      • airpack.tf2
        • airpack.tf2.fileio
        • airpack.tf2.model
    • airpack_scripts
      • airpack_scripts.airt
        • airpack_scripts.airt.run_airt_inference
      • airpack_scripts.pytorch
        • airpack_scripts.pytorch.make_plan_file
        • airpack_scripts.pytorch.run_benchmarks
        • airpack_scripts.pytorch.run_inference
        • airpack_scripts.pytorch.run_training
      • airpack_scripts.tf1
        • airpack_scripts.tf1.make_plan_file
        • airpack_scripts.tf1.run_benchmarks
        • airpack_scripts.tf1.run_inference
        • airpack_scripts.tf1.run_training
      • airpack_scripts.tf2
        • airpack_scripts.tf2.make_plan_file
        • airpack_scripts.tf2.run_benchmarks
        • airpack_scripts.tf2.run_inference
        • airpack_scripts.tf2.run_training

airpack_scripts.tf2.run_training¶

Module Contents¶

airpack_scripts.tf2.run_training.train(data_folder, n_epoch=10)¶

Script used to train the airpack.tf2.model.default_network().

Note

You may modify the parameters in this script to tune the hyperparameters of the airpack.tf2.model.default_network().

Parameters
  • data_folder (Union[str, os.PathLike]) – Location of training data

  • n_epoch (int) – Number of epochs in training process

Returns

Training accuracy

Return type

float

Previous
airpack_scripts.tf2.run_inference
Copyright © 2022, Deepwave Digital, Inc.
Contents
  • airpack_scripts.tf2.run_training
    • Module Contents
      • train