help with training #328

New Issue

2022-10-13T23:32:36+02:00

ggspike commented

2022-10-13 23:32:36 +02:00

(Migrated from github.com)

hi all. I don't understand how model training works.
following the instructions i use the command: "python train.py --name simswap224_test --gpu_ids 0 --batchSize 4 --dataset /content/TrainingData/vggface2_crop_arcfacealign_224 --Gdeep False"

but i get error:
processing Swapping dataset images...
Finished preprocessing the Swapping dataset, total dirs number: 0...
Traceback (most recent call last):
File "train.py", line 159, in
train_loader = GetLoader(opt.dataset,opt.batchSize,8,1234)
File "X:\games\SimSwap-main\data\data_loader_Swapping.py", line 119, in GetLoader
drop_last=True,shuffle=True,num_workers=num_workers,pin_memory=True)
File "C:\Users\Spike\anaconda3\envs\simswap\lib\site-packages\torch\utils\data\dataloader.py", line 266, in init
sampler = RandomSampler(dataset, generator=generator) # type: ignore
File "C:\Users\Spike\anaconda3\envs\simswap\lib\site-packages\torch\utils\data\sampler.py", line 104, in init
"value, but got num_samples={}".format(self.num_samples))
ValueError: num_samples should be a positive integer value, but got num_samples=0

I assume that the program doesn't find pictures for training. where should I put them?

hi all. I don't understand how model training works. following the instructions i use the command: "python train.py --name simswap224_test --gpu_ids 0 --batchSize 4 --dataset /content/TrainingData/vggface2_crop_arcfacealign_224 --Gdeep False" but i get error: processing Swapping dataset images... Finished preprocessing the Swapping dataset, total dirs number: 0... Traceback (most recent call last): File "train.py", line 159, in <module> train_loader = GetLoader(opt.dataset,opt.batchSize,8,1234) File "X:\games\SimSwap-main\data\data_loader_Swapping.py", line 119, in GetLoader drop_last=True,shuffle=True,num_workers=num_workers,pin_memory=True) File "C:\Users\Spike\anaconda3\envs\simswap\lib\site-packages\torch\utils\data\dataloader.py", line 266, in __init__ sampler = RandomSampler(dataset, generator=generator) # type: ignore File "C:\Users\Spike\anaconda3\envs\simswap\lib\site-packages\torch\utils\data\sampler.py", line 104, in __init__ "value, but got num_samples={}".format(self.num_samples)) ValueError: num_samples should be a positive integer value, but got num_samples=0 I assume that the program doesn't find pictures for training. where should I put them?

Sign in to join this conversation.

1 Participants

Notifications

Due Date

No due date set.

Dependencies

No dependencies set.

Reference: CalvinBackup/neuralchen-SimSwap#328