aihwkit.experiments.experiments package

Experiments for aihwkit.

Submodules

• aihwkit.experiments.experiments.base module
  • Experiment
    • Experiment.add_hook()
    • Experiment.clear_hooks()
  • Signals
    • Signals.EPOCH_END
    • Signals.EPOCH_START
    • Signals.EXPERIMENT_END
    • Signals.EXPERIMENT_START
    • Signals.INFERENCE_END
    • Signals.INFERENCE_REPEAT_END
    • Signals.INFERENCE_REPEAT_START
    • Signals.INFERENCE_START
    • Signals.TRAIN_EPOCH_BATCH_END
    • Signals.TRAIN_EPOCH_BATCH_START
    • Signals.TRAIN_EPOCH_END
    • Signals.TRAIN_EPOCH_START
    • Signals.VALIDATION_EPOCH_BATCH_END
    • Signals.VALIDATION_EPOCH_BATCH_START
    • Signals.VALIDATION_EPOCH_END
    • Signals.VALIDATION_EPOCH_START
• aihwkit.experiments.experiments.inferencing module
  • BasicInferencing
    • BasicInferencing.get_data_loader()
    • BasicInferencing.get_dataset_arguments()
    • BasicInferencing.get_dataset_transform()
    • BasicInferencing.get_model()
    • BasicInferencing.inference()
    • BasicInferencing.inferencing_step()
    • BasicInferencing.run()
  • download()
• aihwkit.experiments.experiments.training module
  • BasicTraining
    • BasicTraining.get_data_loaders()
    • BasicTraining.get_dataset_arguments()
    • BasicTraining.get_dataset_transform()
    • BasicTraining.get_optimizer()
    • BasicTraining.run()
    • BasicTraining.train()
    • BasicTraining.training_step()
    • BasicTraining.validation_step()