discriminate the type of the data

assert key.startswith('train') or key.startswith('eval')
if type(value) == torch.Tensor:
　　value = value.item()
  
　　
posted @ 2022-05-28 20:59  呦呦南山  阅读(21)  评论(0)  编辑  收藏  举报