Skip to content

Update distill.py to include device agnostic code for distill_mlp head and distillation_token - #324

Open
vivekh2000 wants to merge 1 commit into
lucidrains:mainfrom
vivekh2000:patch-3
Open

Update distill.py to include device agnostic code for distill_mlp head and distillation_token#324
vivekh2000 wants to merge 1 commit into
lucidrains:mainfrom
vivekh2000:patch-3

Update distill.py to include device agnostic code for `distill_mlp` h…

1027454
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs