ITADN

Batch-size Error

#104ClosedRoyalbx 创建于 2023-11-17
R
Royalbxcommented
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 条评论