5 Commits

Author	SHA1	Message	Date
littletomatodonkey	95d07675d4	fix kldiv input	2021-06-09 08:06:28 +00:00
littletomatodonkey	0343756e46	fix metric	2021-06-03 13:31:25 +00:00
littletomatodonkey	ab4db2acce	support dict output for basemodel	2021-06-03 05:57:31 +00:00
littletomatodonkey	e5d3a2d880	fix distillation arch and model init	2021-06-03 05:30:43 +00:00
littletomatodonkey	ed02b91d26	add distillation function	2021-06-02 08:31:57 +00:00