I want to implement a two-branch architecture with
forward( ) operate differently in train and test mode:
forward(x1, x2)which allows model takes
branch_Bat the same time.
For testing, I want to evaluate each branch separately, such as
How to implement this?