Commit Graph
4 Commits
Author SHA1 Message Date
Kashif Rasul d11578a235 fix final_output tensor shape 2022-06-06 14:11:51 +02:00
Kashif Rasul c99572a6d9 fix args 2022-06-06 11:44:19 +02:00
Kashif Rasul a429cef13d added switch encoder layer 2022-06-06 11:04:43 +02:00
Kashif Rasul 40981abdb4 initial switch transformer 2022-06-06 10:35:36 +02:00