Vivien
Add eval and torch.no_grad (because inference only)
0779f15