U2Fusion
U2Fusion copied to clipboard
请问可以提供训练代码吗?
作者您好,我对U2Fusion的思路很感兴趣,想先试试自己的数据集训练效果,不过看到train_task.py中没有定义main函数,请问方便的话可以提供一下对应的训练代码吗?
训练代码参照main.py