search

AliaksandrSiarohin / first-order-model

This repository contains the source code for the paper First Order Motion Model for Image Animation
https://aliaksandrsiarohin.github.io/first-order-model-website/
MIT License
14.49k stars 3.21k forks source link

Warning Message About Dataloaders, can't train the model on my custom dataset #447

Open aerodiger opened 3 years ago

aerodiger commented 3 years ago

My dataset is only 200 videos, with dimensions 256x256. I get a warning message shown below and it won't let me train the model. I'm using Colab Pro. What should I do? Thanks in advance.

Code: !python run.py --config config/AniVid-Small.yaml --checkpoint data/vox-adv-cpk.pth.tar

Output: Use predefined train-test split. Training... No discriminator in the state-dict. Dicriminator will be randomly initialized /usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py:481: UserWarning: This DataLoader will create 6 worker processes in total. Our suggested max number of worker in current system is 2, which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary. cpuset_checked)) 0it [00:00, ?it/s]

fjesikfjdskl commented 1 year ago

@aerodiger Hello, have you solved this problem?