2020年9月29日
【PyTorch】学习笔记
摘要： 1. 查看PyTorch版本 import torch print(torch.__version__) 2. 模型参数量 model = FPN() num_params = sum(p.numel() for p in model.parameters()) print("num of para 阅读全文
posted @ 2020-09-29 09:39 达可奈特 阅读(363) 评论(0) 推荐(0) 编辑