上传者: 42134144
|
上传时间: 2021-11-10 15:53:53
|
文件大小: 22.02MB
|
文件类型: -
多任务学习
此仓库旨在在PyTorch中实施几种多任务学习模型和培训策略。 该代码库是对以下工作的补充:
, ,Wouter Van Gansbeke,Marc Proesmans,Dengxin Dai和Luc Van Gool。
, 和Luc Van Gool。
在可以找到有关多任务学习的最新作品列表。
安装
该代码与最新的Pytorch版本一起运行,例如1.4。 假设 ,最重要的软件包可以按以下方式安装:
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
conda install imageio scikit-image # Image operations
conda install -c conda-forge opencv # OpenCV
cond