mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
fixing documentation for unfinished module ordering function
This commit is contained in:
1 parent
f17192a93e
commit
c7911281c3
1 file changed
+9
-14
+9
-14
@@ -15,11 +15,7 @@ class GraphInputs:
|
||||
|
||||
|
||||
class Graph(nn.Module):
|
||||
"""_summary_
|
||||
|
||||
Args:
|
||||
nn (_type_): _description_
|
||||
"""
|
||||
"""_summary_ """
|
||||
|
||||
def __init__(self, modules_config: dict) -> None:
|
||||
super().__init__()
|
||||
@@ -34,17 +30,16 @@ class Graph(nn.Module):
|
||||
module_dict["meta_data"]["in_dim"] = in_dim
|
||||
self.modules[module_name] = module(**module_dict["meta_data"])
|
||||
|
||||
def get_module_order(self, modules_config: list):
|
||||
"""Generates a graph to determine the order in which the graph should execute
|
||||
def get_module_order(self, modules_config: dict) -> list:
|
||||
"""Generates a graph to determine the order in which the modules should execute
|
||||
|
||||
Args:
|
||||
modules_config (list): _description_
|
||||
modules_config (dict): module definitions that make up the network
|
||||
|
||||
Returns:
|
||||
_type_: _description_
|
||||
list: ordering of the module names that should be executed
|
||||
"""
|
||||
# raise NotImplementedError
|
||||
return modules_config
|
||||
raise NotImplementedError
|
||||
|
||||
def forward(self, x):
|
||||
"""_summary_
|
||||
@@ -52,6 +47,6 @@ class Graph(nn.Module):
|
||||
Args:
|
||||
x (_type_): _description_
|
||||
"""
|
||||
# raise NotImplementedError
|
||||
for name in self.module_order:
|
||||
x = self.modules[name](x)
|
||||
# for name in self.module_order:
|
||||
# x = self.modules[name](x)
|
||||
raise NotImplementedError
|
||||
Reference in new issue
Block a user