search

Haochen-Wang409 / U2PL

[CVPR'22 & IJCV'24] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels & Using Unreliable Pseudo-Labels for Label-Efficient Semantic Segmentation
Apache License 2.0
436 stars 61 forks source link

resnet101.pth文件寻找不到 #101

Closed bulindun closed 1 year ago

bulindun commented 1 year ago

作者您好，冒昧打扰。我在下载了resnet101.pth文件后，将其放置在u2pl/models文件夹下，resnet.py里面已经有了resnet101.pth的路径，但是train.sh运行时仍报错无法找到resnet101.pth文件： resnet.py中的程序： 1 训练的报错信息： 2

Haochen-Wang409 commented 1 year ago

直接在 resnet.py 中把路径改为您保存的路径即可

bulindun commented 1 year ago

不好意思再次打扰。CUDNN没有初始化的问题 4 请问这是什么情况？ 我的torch1.8.1，torchvision0.9.1, cuda10.2.89, cudnn7.6.5

Haochen-Wang409 commented 1 year ago

一般的 cuda 训练程序是否能训练起来？比如在 MNIST 数据集用 CUDA 训练一个 ResNet-18