Skip to content

osama-usuf/layer-ensemble-averaging

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Descriptions of provided code files:

  • dataset.py - Defines the multi-task variant of the Yin-Yang dataset. Apapted from code by original authors.
  • train_multitask_ewc.py - A PyTorch implementation of neural network training for the investigated network architecture, multi-task classification task with elastic weight consolidation.
  • layer_ensemble_averaging.py - Defines helper functions for the core layer ensemble averaging framework and hardware neural network deployment for inference. Requires primitives from daffodil-lib and daffodil-app (will be publicly released soon).

NOTE: Code for quantizing software solutions is not included as it is based on the publicly available BRECQ implementation: https://github.com/yhhhli/BRECQ

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages