machine learn
xmy21sunshine
这个作者很懒,什么都没留下…
展开
-
使用thop统计模型flops和#param.
from thop import profile device = 'cuda:1' model = Transformer().to(device) input = torch.randn(1, 201, 192).to(device) macs, params = profile(model, inputs=(input,)) print("macs\params:",macs,params)原创 2022-04-20 18:41:31 · 1380 阅读 · 0 评论 -
多GPU运行torch
34 # model = BiLSTM_Attention(128, 64, 2, 40, True) 35 model = Transformer() 36 model = nn.DataParallel(model, device_ids=[0, 1, 2, 3]).cuda() # 增加改行代码后即可运行多GPU环境 37 optimizer = torch.optim.Adam(model.parameters(), lr=opt.init_lr) ...原创 2022-04-19 09:47:50 · 1556 阅读 · 0 评论 -
Windows安装Pycocotools(Microsoft Visual C++ 14.0 or greater is required.报错提示解决
【深度学习】Windows安装Pycocotools(Microsoft Visual C++ 14.0 or greater is required.报错提示解决 https://www.cnblogs.com/Flat-White/p/15123308.html原创 2021-10-15 00:00:38 · 86 阅读 · 0 评论 -
arima 基本资料转载
arima视频:https://www.bilibili.com/video/BV1Ks411V7W4?p=2 arima视频对应的资料链接:https://blog.csdn.net/weixin_41988628/article/details/83149849 ARIMA模型实例https://blog.csdn.net/sunnyxidian/article/details/929465...原创 2020-04-20 21:56:05 · 168 阅读 · 0 评论