Skip to content

请问下载的模型和数据需要放在哪里呢? #5

@SilentZhang

Description

@SilentZhang

我是一个初学者,请问下载的模型和数据需要放在哪个目录下呢

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions