Open Peter-weng opened 1 year ago
Your dataset format is wrong. There is a train_own_datasets parameter in train.py needs to be set.
fine,i set train_own_datasets = True,which dataset format wrong? i used DDP in 8*3090 but i think it slow about dataloader,would you like to add cache image function in training like yolov5?
Can you start training now? I set two data formats, corresponding to different data sets. The original data format corresponds to the omniglot dataset. I don't set cache, but I think you can improve num_ works.
yeah, i can train now and finish the cache function,thanks for your reply
3q
hello,i meet some problem when i frist time run the code. error:
i think the self.types may be the size of labels,so i change the code and work for me like this,
maybe is difficult version env make this problem?