Hi,
I am trying to train face2video on my custom video.
When I run python train.py --blink_path '/content/drive/MyDrive/mydata/Record_2023_05_31_11_10_51_101.csv' --name train3 --model pose2vid --dataroot ./datasets/train3/ --netG local --ngf 32 --num_D 3 --tf_log --niter_fix_global 0 --label_nc 0 --no_instance --save_epoch_freq 2 --lr=0.0001 --resize_or_crop resize --no_flip --verbose --n_local_enhancers 1
after 1 epoch:
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/drive/.shortcut-targets-by-id/1oeGKHgJCIZ8MTmTxo6L0325T8Hm8eihd/FACIAL/face2vid/data/aligned_pair_dataset.py", line 83, in __getitem__ a = torch.FloatTensor([blink[i]]) IndexError: index 9 is out of bounds for axis 0 with size 9
train3 folder consists of 4 folders:
train_A: all frames
train_B: all frames - the first 25s
test_A and test_B are the same (output of previous stage)
What should I do? please help me
@zhangchenxu528
Hi, I am trying to train face2video on my custom video. When I run
python train.py --blink_path '/content/drive/MyDrive/mydata/Record_2023_05_31_11_10_51_101.csv' --name train3 --model pose2vid --dataroot ./datasets/train3/ --netG local --ngf 32 --num_D 3 --tf_log --niter_fix_global 0 --label_nc 0 --no_instance --save_epoch_freq 2 --lr=0.0001 --resize_or_crop resize --no_flip --verbose --n_local_enhancers 1after 1 epoch:
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp> data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/drive/.shortcut-targets-by-id/1oeGKHgJCIZ8MTmTxo6L0325T8Hm8eihd/FACIAL/face2vid/data/aligned_pair_dataset.py", line 83, in __getitem__ a = torch.FloatTensor([blink[i]]) IndexError: index 9 is out of bounds for axis 0 with size 9train3 folder consists of 4 folders: train_A: all frames train_B: all frames - the first 25s test_A and test_B are the same (output of previous stage) What should I do? please help me @zhangchenxu528