zero.module

functions

call(module, *args, **kwargs)	Move arguments to module’s device and call the module with these arguments.
ecall(module, *args, **kwargs)	Call the module (torch.no_grad() + module.eval() + input.to(device)).
evaluation(*modules)	Context-manager for models evaluation.