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

finish generate_sequence.py

Browse Source
This commit is contained in:
hofee
2024-08-22 20:27:21 +08:00
parent ff3e89b17c
commit 7cd1954a25
7 changed files with 121 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
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")
@stereotype.pipeline("nbv_reconstruction_pipeline", comment="should be tested")
class NBVReconstructionPipeline(nn.Module):
def __init__(self, config):
super(NBVReconstructionPipeline, self).__init__()