Initial commit
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
//! Knowledge distillation framework for model compression
|
||||
|
||||
pub mod knowledge_distillation;
|
||||
pub mod tensor_extensions;
|
||||
|
||||
pub use knowledge_distillation::{
|
||||
AttentionDistillationResult, AttentionTransferConfig, CrossModalDistillationResult,
|
||||
DistillationConfig, DistillationLoss, DistillationMethod, DistillationResult,
|
||||
FeatureDistillationResult, FeatureMatchingConfig, KnowledgeDistiller, MultiTeacherConfig,
|
||||
MultiTeacherDistillationResult, OnlineDistillationResult, ProgressiveDistillationConfig,
|
||||
StudentModel, TeacherModel,
|
||||
};
|
||||
pub use tensor_extensions::{DistillationTensorOps, DistillationUtils};
|
||||
Reference in New Issue
Block a user