Pytorch Dict Object Is Not Callable