Train tgnet_fps and find the error
Hello, when I train the tgnet_fps model, I got following error
Traceback (most recent call last):
File "start_train.py", line 51, in
How to solve this issue? By the way , I use the challange dataset.
My base_name_train_fold.txt belike 00OMSZGW_lower_lower_sampled_points.npy 00OMSZGW_upper_upper_sampled_points.npy 01328DDN_lower_lower_sampled_points.npy...... Is that right? or it should contain input_data_dir_path
I modify the code ,change the parameter of shuffle from true to false,solve this issue。 point_loader = DataLoader( DentalModelGenerator( config["input_data_dir_path"], aug_obj_str=config["aug_obj_str"], split_with_txt_path=config["train_data_split_txt_path"] ), shuffle=False, batch_size=config["train_batch_size"], collate_fn=collate_fn ) But my data path is not set right, the train_set=0, validation_set=0. My input_data_dir_path is set to the folder where data is stored,and txt file belike 00OMSZGW_lower_lower_sampled_points.npy 00OMSZGW_upper_upper_sampled_points.npy 01328DDN_lower_lower_sampled_points.npy......
I solve the problem ,txt file should be like 01328DDN
I solve the problem ,txt file should be like 01328DDN
What's your video card? How long did it take to train tgnet_fps?