Running pre-training script with point contrast about 3dtrans HOT 3 CLOSED

Update:
The code crashes in the following block:
# build unsupervised dataloader datasets, dataloaders, samplers = build_unsupervised_dataloader( dataset_cfg=cfg.DATA_CONFIG, class_names=cfg.CLASS_NAMES, batch_size=args.batch_size, root_path=cfg.DATA_CONFIG.DATA_PATH, dist=dist_train, workers=args.workers, logger=logger, merge_all_iters_to_one_epoch=args.merge_all_iters_to_one_epoch )

Found in the tools/train_pointcontrast.py file

from 3dtrans.

Update:
The code seems to crash on this specific code block in once_semi_dataset.py
On line 64 and 65.
The script sometimes manages to load the once_raw_data_small.pkl file and sometimes doesn't.

Do you have any idea why this might be happening @BOBrown?

from 3dtrans.

Update.
Fixed the problem, it was due to a lack of GPU resources
Thanks for help :)

from 3dtrans.