-
Notifications
You must be signed in to change notification settings - Fork 12
Description
我是一个初学者,请问下载的模型和数据需要放在哪个目录下呢
We use one dataset supported by Baidu Aistdio at https://aistudio.baidu.com/aistudio/datasetdetail/121039
You can use the flowing links to download the Pretrained model and the datasets.
Pretrained model: https://pan.baidu.com/s/1QmCGhwpKNtdKtYoGU0YK3w?pwd=a2jh
Datasets: https://pan.baidu.com/s/1l2mQ3FLx_NAPJEh30cOeZA?pwd=05bl
现在执行命令报错,感觉是找不到这些文件。
(llmfy) PS D:\DEV\repos\HandWritingEraser-Pytorch> python main.py --data_root ./datasets/HandWriting --loss_type focal_loss --gpu_id 0 --batch_size 4 --ckpt checkpoints/best_deeplabv3plus_resnet50_os16.pth --test_only --save_val_results Backbone HRNet Pretrained weights at: ./checkpoints/hrnetv2_32_model_best_epoch96.pth, only usable for HRNetv2-32 Setting up a new session... Device: cuda ./datasets/HandWriting images number: 1 ./datasets/HandWriting images number: 1 Traceback (most recent call last): File "D:\DEV\repos\HandWritingEraser-Pytorch\main.py", line 347, in <module> main() File "D:\DEV\repos\HandWritingEraser-Pytorch\main.py", line 195, in main train_loader = data.DataLoader( File "C:\Users\zhang\.conda\envs\llmfy\lib\site-packages\torch\utils\data\dataloader.py", line 383, in __init__ sampler = RandomSampler(dataset, generator=generator) # type: ignore[arg-type] File "C:\Users\zhang\.conda\envs\llmfy\lib\site-packages\torch\utils\data\sampler.py", line 165, in __init__ raise ValueError( ValueError: num_samples should be a positive integer value, but got num_samples=0