search

LucasKre / dilated_tooth_seg_net

DilatedToothSegNet: Tooth Segmentation Network on 3D Dental Meshes Through Increasing Receptive Vision
https://link.springer.com/article/10.1007/s10278-024-01061-6
21 stars 5 forks source link

NameError: name 'fps' is not defined #2

Open littlezhang231 opened 5 months ago

littlezhang231 commented 5 months ago

Hello I met this question in running this code he value of the num_workers argumenttonum_workers=11in theDataLoader` to improve performance. Sanity Checking DataLoader 0: 0%| | 0/2 [00:00<?, ?it/s]Traceback (most recent call last): File "/mnt/dilated_tooth_seg_net-main/train_network.py", line 102, in trainer.fit(model=model, train_dataloaders=train_dataloader, val_dataloaders=val_dataloader, ckpt_path=args.ckpt) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 545, in fit call._call_and_handle_interrupt( File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/call.py", line 44, in _call_and_handle_interrupt return trainer_fn(*args, kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 581, in _fit_impl self._run(model, ckpt_path=ckpt_path) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 990, in _run results = self._run_stage() File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 1034, in _run_stage self._run_sanity_check() File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/trainer.py", line 1063, in _run_sanity_check val_loop.run() File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/utilities.py", line 181, in _decorator return loop_run(self, *args, kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/evaluation_loop.py", line 134, in run self._evaluation_step(batch, batch_idx, dataloader_idx, dataloader_iter) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/loops/evaluation_loop.py", line 391, in _evaluation_step output = call._call_strategy_hook(trainer, hook_name, step_args) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/trainer/call.py", line 309, in _call_strategy_hook output = fn(args, kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/lightning/pytorch/strategies/strategy.py", line 403, in validation_step return self.lightning_module.validation_step(*args, kwargs) File "/mnt/dilated_tooth_seg_net-main/models/dilated_tooth_seg_network.py", line 109, in validation_step pred = self.model(x, pos) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(*args, kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, *kwargs) File "/mnt/dilated_tooth_seg_net-main/models/dilated_tooth_segnetwork.py", line 63, in forward x1, = self.dilated_edge_graph_conv_block1(x, pos, cd=cd) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl return self._call_impl(args, kwargs) File "/root/miniconda3/envs/myconda/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl return forward_call(*args, *kwargs) File "/mnt/dilated_tooth_seg_net-main/models/layer.py", line 229, in forward idx_fps = fps(pos.reshape(B N, -1)[idx_l], self.k).long() NameError: name 'fps' is not defined

How can I solve it It happpens after processing data