hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change config and ensure pipeline works fine

Browse Source
This commit is contained in:
hofee 2024-09-13 14:12:35 +00:00
parent bdd70323a3
commit d80d0ea79d
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app_train.py
View File

@ -5,4 +5,4 @@ from PytorchBoot.runners.trainer import DefaultTrainer
class TrainApp:
@staticmethod
def start():
DefaultTrainer("configs/train_config.yaml").run()
DefaultTrainer("configs/server/train_config.yaml").run()

3
core/pipeline.py
View File

@ -5,7 +5,7 @@ import PytorchBoot.stereotype as stereotype
from PytorchBoot.factory.component_factory import ComponentFactory
from PytorchBoot.utils import Log
@stereotype.pipeline("nbv_reconstruction_pipeline", comment="should be tested")
@stereotype.pipeline("nbv_reconstruction_pipeline")
class NBVReconstructionPipeline(nn.Module):
def __init__(self, config):
super(NBVReconstructionPipeline, self).__init__()
@ -72,7 +72,6 @@ class NBVReconstructionPipeline(nn.Module):
pose_feat_seq_list = []
for scanned_pts,scanned_n_to_1_pose_9d in zip(scanned_pts_batch,scanned_n_to_1_pose_9d_batch):
print(scanned_n_to_1_pose_9d.shape)
scanned_pts = scanned_pts.to(best_to_1_pose_9d_batch.device)
scanned_n_to_1_pose_9d = scanned_n_to_1_pose_9d.to(best_to_1_pose_9d_batch.device)
pts_feat_seq_list.append(self.pts_encoder.encode_points(scanned_pts))