- 博客(3)
- 收藏
- 关注
原创 CUDA11 + mmsegmentation(swin-T)
1.创建虚拟环境硬件及系统:RTX3070 + Ubuntu20.043070只能用cuda11+的版本进行加速,官方教程中的配置都是cuda10。选择使用pytorch1.8.1及cuda11.1。想使用GPU加速则必须安装mmcv-full库,该库与mmcv不兼容(不要用pip install mmcv)。mmcv-full版本可以到https://download.openmmlab.com/mmcv/dist/内查找确认,我使用了1.3.8版本。conda create -n lab py
2021-07-13 15:37:09 2163 5
原创 Transformer代码个人理解(Pytorch)
Transformer代码个人理解(Pytorch)刚开始接触深度学习的内容,以下代码原文可直接执行。对于Transformer的理解也比较肤浅,如有不足之处还请批评指正。#!/usr/bin/python# -*- coding: UTF-8 -*-import numpy as npimport torchimport torch.nn as nnimport torch.nn.functional as F# nn.与nn.functional.的实际功能是相同的,前者是后者的类封装
2020-12-27 00:46:06 556 1
原创 阿里云Graph-Learn框架安装指南
阿里云Graph-Learn框架安装指南1.GL框架简述Graph-Learn(GL,原AliGraph) 是面向大规模图神经网络的研发和应用而设计的一款分布式框架, 它从实际问题出发,提炼和抽象了一套适合于当下图神经网络模型的编程范式, 并已经成功应用在阿里巴巴内部的诸如搜索推荐、网络安全、知识图谱等众多场景。@article{zhu2019aligraph,title={AliGraph: a comprehensive graph neural network platform},auth
2020-12-09 00:07:26 695 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人