TengdaHan / CoCLR

[NeurIPS'20] Self-supervised Co-Training for Video Representation Learning. Tengda Han, Weidi Xie, Andrew Zisserman.
Apache License 2.0
286 stars 32 forks source link

program stuck when creating data loader #29

Closed youcaiSUN closed 3 years ago

youcaiSUN commented 3 years ago

Hi tengda, I want to reproduce the result of infonce on UCF101. However, when I ran the code (in the section of "InfoNCE pretrain on UCF101-RGB", I did not change anything), the program stuck when creating data loader. Could you help me to figure out this problem? Thanks very much! Below is the console output:

`... module.encoder_k.0.Mixed_5c.branch3.1.bn.bias False module.encoder_k.2.weight False module.encoder_k.2.bias False module.encoder_k.4.weight False module.encoder_k.4.bias False \=================================

TransformController: [<utils.augmentation.TwoClipTransform object at 0x7f8e866e6850>, <utils.augmentation.OneClipTransform object at 0x7f8e866e6990>] with weights: [0.5, 0.5] Loading data for "train" mode Loading LMDB from /data1/Deep Learning/Dataset/UCF101/ucf101_frame.lmdb, split:1 Frame Dataset from "/data1/Deep Learning/Code/CoCLR/dataset/../process_data/data/ucf101" has #class 101 filter out too short videos ... Creating data loaders for "train" mode `