代码
zhh_hhz
这个作者很懒,什么都没留下…
展开
-
maven常用指令
maven入门-常用指令原创 2022-08-26 11:34:48 · 143 阅读 · 1 评论 -
配置st-metanet环境(DGL+MXNET)
dgl0.6.0下载地址:https://pypi.tuna.tsinghua.edu.cn/packages/13/c4/d1b00c7cbb186b6d41bbe9f13037409dd05a423b04191166582d98251454/dgl_cu101-0.6.0-cp36-cp36m-manylinux1_x86_64.whl#sha256=e86b8dd50b85d17b14d7aee080c4724c368fc352bc8a75c93222843c048adc49原创 2021-07-19 14:41:34 · 344 阅读 · 0 评论 -
Pytorch多GPU训练
参考博客:https://blog.csdn.net/daydayjump/article/details/81158777使用方法device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")定义device,其中需要注意的是“cuda:0”代表起始的device_id为0,如果直接是“cuda”,同样默认是从0开始。可以根据实际需要修改起始位置,如“cuda:1”。model = Model()if torch.原创 2020-05-11 20:05:25 · 248 阅读 · 0 评论 -
RACNN学习和使用
RACNN源代码来自: https://github.com/klrc/RACNN-pytorch我的数据集形式二分类数据集,分为train test validaton三个子文件夹,每个子文件夹又分为两个文件,分别存放一类图片,如下图所示getitem def __getitem__(self, index): img = cv2.imread(self._img...原创 2020-05-11 19:58:45 · 2572 阅读 · 3 评论 -
李代数骨骼行为预测-代码学习
Argparsefrom argparse import ArgumentParserparser = Argumentparser()paser.add_argument("--dataset",,,)args = parser.parse_args()## 使用dataset = args.datasetconfig自定义TrainConfig类, 里面存储训练需要的配置参数...原创 2020-05-07 19:37:55 · 330 阅读 · 1 评论