facebookresearch/SlowFast

Runtime error

Open

#172 ouverte le 20 avr. 2020

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)Python (1 167 forks)batch import
help wanted

Métriques du dépôt

Stars
 (6 064 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

Thank you very much for your excellent work. I wanted to train with the ava data set, but it went wrong: RuntimeError: Caught RuntimeError in DataLoader worker process 0. RuntimeError: invalid argument 0: Sizes of tensors must match the except in dimension 0. Got 3 and 2 dimension in the at 1 / pytorch aten/SRC/TH/generic/THTensor CPP: 689 My command line is: python tools/run_net.py --cfg configs/AVA/c2/SLOWFAST_32x2_R101_50_50.yaml DATA.PATH_TO_DATA_DIR /data/ava/videos NUM_GPUS 4 TRAIN.BATCH_SIZE 8 Is there anything wrong with my data set path setting?How should you modify it?Thank you very much! My data set and configuration files were downloaded according to your command.

Guide contributeur