Improving KD via reducing regularization.
install requirments.txt
run python train.py with the following arguments :
- teacher
- student
- learning rate
- pruning rate
- pruning mode
- batch size
- epochs
- optimizer
- temperature
- etc ..
-
Notifications
You must be signed in to change notification settings - Fork 0
mube1/thesis
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Thesis implementation
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published