2020年12月1日
torch.chunk()函数
摘要： 将tensor按dim方向分割成chunks个tensor块，返回的是一个元组。 示例： 阅读全文
posted @ 2020-12-01 22:41 Picassooo 阅读(2982) 评论(0) 推荐(0)
Python __call__()方法【转】
摘要： Python __call__()方法 阅读全文
posted @ 2020-12-01 18:40 Picassooo 阅读(76) 评论(0) 推荐(0)
Pytorch 统计模型参数量 param.numel()【转】
摘要： Pytorch 统计模型参数量 param.numel() total_params = 0 trainable_params = 0 for param in self.lane_head.parameters(): param_count = param.numel() # 获取单个参数的元素个 阅读全文
posted @ 2020-12-01 16:30 Picassooo 阅读(237) 评论(0) 推荐(0)
Python __import__() 函数 【转】
摘要： Python __import__() 函数 阅读全文
posted @ 2020-12-01 11:27 Picassooo 阅读(76) 评论(0) 推荐(0)
Python getattr() 函数【转】
摘要： Python getattr() 函数 阅读全文
posted @ 2020-12-01 11:25 Picassooo 阅读(82) 评论(0) 推荐(0)