Skip to content

Latest commit

 

History

History
12 lines (7 loc) · 215 Bytes

pytorch_tools.md

File metadata and controls

12 lines (7 loc) · 215 Bytes

pytorch_note

  1. 查看模型名称:

    for name, param in model.named_parameters():
        if param.requires_grad:
            print('hbsun-debug', name, param.data.shape)