median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning
MIT License
1.94k stars 181 forks source link

v1.1.6文档 #33

Closed DonnyLiu7 closed 1 year ago

DonnyLiu7 commented 1 year ago

作者你好,不知道是否有v1.1.6的最新文档呢?我看新加的一些参数例如cfg等等都没有完整的例子与介绍,文档还是原来的版本

Baijiong-Lin commented 1 year ago

v1.1.6的文档还需要一些时间完善,如果你对一些参数有不清楚的,可以直接向我们提问

DonnyLiu7 commented 1 year ago

我想加载自己的数据集,multi_inputs = True,存放格式是这样的: -dataset -task_1 -train -val -task_2 -train -val 请问如何在v1.1.6的配置文件中的dataloader部分正确的加载数据?

Baijiong-Lin commented 1 year ago

你可以参考OfficeDataset.py来定制你的dataset,这个dataset文件定义了每个task在每个mode(train,val,test)下的dataset

DonnyLiu7 commented 1 year ago

非常感谢