Fixed it! In UNK_C6: DecoderBlock, feed_forward, I had used tl.Dropout(dropout, mode), instead of tl.Dropout(dropout, mode = mode) - thusly giving parameters in the wrong order. Thank you!
Fixed it! In UNK_C6: DecoderBlock, feed_forward, I had used tl.Dropout(dropout, mode), instead of tl.Dropout(dropout, mode = mode) - thusly giving parameters in the wrong order. Thank you!