Commit Graph

5 Commits

Author SHA1 Message Date
Kashif Rasul 7b5cd6052f fix for input_size > 1 2022-10-17 12:00:28 +02:00
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