Solved by Abdelrahman_Mohamed in post #8 
              
              
                I solved the problem but there is something I can’t understand , in this block of code the problem is solved and the generated average model work as expected 
model_1=torch.load('Ensemble_Modelfold-1.pt')
model_2=torch.load('Ensemble_Modelfold-2.pt')
model_3=torch.load('Ensemble_Modelfold-3.pt')
mod…