Signed-off-by: Kaihang Jiang <kaihangj@login-lyris02.lyris.clusters.nvidia.com>
torch.cuda.Event
torch.Event
load_weights