Skip to content

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

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

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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