机器学习
Aistra
这个作者很懒,什么都没留下…
展开
-
torch.scatter 或torch.gather运行速度太慢,怎么解决
可以尝试在cpu上进行运算,在上传到gpu。这类运算似乎在gpu上特别的慢原创 2021-12-30 00:07:27 · 1086 阅读 · 0 评论 -
ROCM指定gpu训练
ROCR_VISIBLE_DEVICESIt is possible to rearrange or isolate the collection of ROCm GPU/GCD devices that are available on a ROCm platform. This can be achieved at the start of an application by way of ...原创 2020-05-01 17:36:38 · 661 阅读 · 0 评论 -
利用stacking 提供银行精准营销解决方案(深度之眼比赛)
import pandas as pdimport numpy as npimport warningsfrom tqdm import tqdm%matplotlib inlinefrom sklearn.preprocessing import LabelEncoderfrom scipy import sparsefrom sklearn.model_selection im...原创 2019-07-05 13:26:58 · 1183 阅读 · 0 评论