To convert the model to Relay I had to “fix” begin, end and final_output values inside the code
begin, end and final_output
begin: [(? + -1), 0, 0] -> [0 ,0, 0] end: [((? + -1) + 1), ?, 512] -> [1, 1, 512] final_output: [?, 512] -> [1, 512]
here and here