网络模型很难迁移

lidq92 / MDTVSFA

[official] Unified Quality Assessment of In-the-Wild Videos with Mixed Datasets Training (IJCV 2021)

MIT License

83 stars 16 forks source link

网络模型很难迁移 #12

Closed upupbo closed 2 years ago

upupbo commented 2 years ago

我只想训练konvid-1k,但是你们的网络里面有个key，导致我必须要建立一个key

lidq92 commented 2 years ago

我现在提供的代码可以只训练konvid-1k。我不清楚难迁移什么意思。

如果你要去掉key也是可以的，简单改一下就行了。

upupbo commented 2 years ago

您好，请问self.lm = nn.Sequential(OrderedDict([(dataset, nn.Linear(1, 1)) for dataset in mapping_datasets]))和 for d, dataset in enumerate(mappingdatasets): torch.nn.init.constant(self.lm.modules[dataset].weight, scale[dataset]) torch.nn.init.constant(self.lm._modules[dataset].bias, m[dataset]) 以及aligned_score[d][i] = self.lm._modulesKCL[i] 如何去掉key，我们想直接用VSFA的框架运行应该怎么改，或者您可以提供一版不带key的嘛

lidq92 commented 2 years ago

你直接在VSFA的基础上加上MDTVSFA的后面结构再换成MDTVSFA的loss就可以了。这个改起来很简单，我相信你自己改也没问题，我有我的事情要忙，谢谢。