I am trying to replicate this paper, with the latest version of TensorFlow (2.18.0) but I am getting ValueError: None Values not supported. I think the problem is with the loss function execution. Any help will be highly appreciated. Thank you. My code is as follows: code_part1
code_part2
code_part3
