Many Thanks for your help! When i run the training model the file is saving the tensor which consist of 16 values. However the values seem to get over written at each step and the final CSV final only has 3 values. Any idea why I am not getting all the values? Thanks