aquvitae icon indicating copy to clipboard operation
aquvitae copied to clipboard

Knowledge Distillation Toolkit

Results 1 aquvitae issues
Sort by recently updated
recently updated
newest added

lb.shape: (640, 1280) main() File "tools/train_aquvitae.py", line 85, in main student = Dist( File "/home/xxx/anaconda3/envs/tp/lib/python3.8/site-packages/aquvitae/dist.py", line 28, in dist return _torch_dist( File "/home/xxx/anaconda3/envs/tp/lib/python3.8/site-packages/aquvitae/dist.py", line 118, in _torch_dist loss = algo.teach_step(x,...