Batch-size Error
I get the error, and don't know how to solve it.
python -m pytorch_fid D:/Code/dcgan/dataset/cargo D:/Code/PyTorch-GAN/implementations/wgan_gp/cargo_generat2
Warning: batch size is bigger than the data size. Setting batch size to data size
Traceback (most recent call last):
File "D:\Anaconda\Anaconda3\envs\GAN\lib\runpy.py", line 193, in _run_module_as_main
"__main__", mod_spec)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\runpy.py", line 85, in _run_code
exec(code, run_globals)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\pytorch_fid\__main__.py", line 3, in <module>
pytorch_fid.fid_score.main()
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\pytorch_fid\fid_score.py", line 317, in main
num_workers)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\pytorch_fid\fid_score.py", line 260, in calculate_fid_given_paths
dims, device, num_workers)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\pytorch_fid\fid_score.py", line 244, in compute_statistics_of_path
dims, device, num_workers)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\pytorch_fid\fid_score.py", line 228, in calculate_activation_statistics
act = get_activations(files, model, batch_size, dims, device, num_workers)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\pytorch_fid\fid_score.py", line 126, in get_activations
num_workers=num_workers)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\torch\utils\data\dataloader.py", line 274, in __init__
batch_sampler = BatchSampler(sampler, batch_size, drop_last)
File "D:\Anaconda\Anaconda3\envs\GAN\lib\site-packages\torch\utils\data\sampler.py", line 219, in __init__
"but got batch_size={}".format(batch_size))
ValueError: batch_size should be a positive integer value, but got batch_size=0
关闭于 2024-02-07 5 条评论