论文原文
1.克隆仓库
2. 创建环境
创建并激活环境
conda create -n vipt python=3.8
conda activate vipt
安装Pytorch和对应的cuda
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
安装需要的包
bash install_vipt.sh
3. 数据集准备
我使用的LasHeR数据集。
--datasets/LasHeR
-- trainingset
|-- 1boygo
|-- 1handsth
|-- ........
--testingset
|--

本文记录了ViPT代码的运行过程,包括克隆仓库、创建环境、准备数据集、路径设置和训练。在训练过程中遇到数据读取错误和PyTorch关于输出形状的警告,通过修改代码解决了问题,使得模型能够成功运行。
最低0.47元/天 解锁文章
817

被折叠的 条评论
为什么被折叠?



