Open Scurry61 opened 2 years ago
maybe you can make TrainImgLoader = DataLoader(train_dataset, args.batch_size, shuffle=True, num_workers=16, drop_last=True) reduce the number of num_workers
I also encountered the same problem as you, how did you solve it, thank you for your reply
thanks for your great work! when i doing the first step,"python main.py --attention_weights_only True",there is always an error with loop output “creating new summary file loading model ./pretrained_model/pretrained_model_sceneflow.ckpt start at epoch 0” and then,no enough storage,i am wondering Is it a problem with the laptop memory? If i want to use the pretrained model,how can i finetune it in my own dataset? thanks for your reply