Use totem instead of torch.Tester

For dev purpose only, until totem is published.