-
Notifications
You must be signed in to change notification settings - Fork 626
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
dataloader question when ./train_mixed.sh #49
Comments
The dataloader isn't able to find the training dataset. Either use the directory structure in the Readme or change the dataset_path variable in datasets.py |
Thanks for answer. I still have some questions: 1.Does the command (./train_mixed.sh) run only when the four datasets all in the directory? If I only have two datasets(such as Sintel and DAVIS), can I train the network? 2. If I want to train another dataset(my private dataset), what should I do? Thanks!!! |
You can modify the function Line 199 in d3f3840
to
to train only using sintel data. To train on another dataset, you will need to write your own subclass of FlowDataset |
Thanks again for your answer! I modified the function to train only using Sintel dataset. However, there exists another question during training: |
In the training command, you need to change the flag to specify the validation dataset, from |
Thanks again for your answer! If I want to save all the output flow images in a directory (just flow images, not an original frame on the top and a flow image on the below in the output of demo.py), what should I do? thanks a lot! |
File "train.py", line 247, in
train(args)
File "train.py", line 150, in train
train_loader = datasets.fetch_dataloader(args)
File "core/datasets.py", line 230, in fetch_dataloader
train_loader = data.DataLoader(train_dataset, batch_size=args.batch_size,
File "/data/zzl/anaconda3/envs/raft/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 224, in init
sampler = RandomSampler(dataset, generator=generator)
File "/data/zzl/anaconda3/envs/raft/lib/python3.8/site-packages/torch/utils/data/sampler.py", line 95, in init
raise ValueError("num_samples should be a positive integer "
ValueError: num_samples should be a positive integer value, but got num_samples=0
The text was updated successfully, but these errors were encountered: