训练卡在加载pth文件

Linfeng-Tang / SwinFusion

This is official Pytorch implementation of "SwinFusion: Cross-domain Long-range Learning for General Image Fusion via Swin Transformer"

161 stars 20 forks source link

Open SadInSummer opened 1 month ago

SadInSummer commented 1 month ago

想直接训练，但是卡在这个地方，代码只把并行运算关掉了，因为单卡，是因为我的算力不够的原因吗？

Linfeng-Tang commented 1 month ago

应该不是为什么需要加载预训练的文件呢从头开始训练可以运行吗？

SadInSummer commented 1 month ago

应该不是为什么需要加载预训练的文件呢从头开始训练可以运行吗？

这个如何不加载预训练模型呢？看了半天没看明白，还请指教，十分感谢