记录
Conv3d
Dev
展开
-
bug: matplotlib 画混淆矩阵,少一列 办法:改版本 3.1.1 为 3.0.3
pip install matplotlib==3.0.3原创 2019-11-27 15:10:48 · 802 阅读 · 2 评论 -
pytorch 使用记录
1 用到在用log_softmax +cross_entropy 在做多分类的时候: 损失函数的输入如下:label 序号应该从0 开始 定义。 2 监控内存使用, 防止内存泄露(memory leak) import gc import resource gc.collect() max_mem_used = resource.getrusage(resource.RUSAGE...原创 2019-07-04 09:42:19 · 321 阅读 · 0 评论