Commit 6229f294 authored by lr119628's avatar lr119628
Browse files

[update] fixed node count

parent 162534be
......@@ -565,10 +565,10 @@ class Net_${networkInstruction?index}(gluon.HybridBlock):
self.candidate_complexities[name] = operation.get_complexity()
self.out = nn.Dense(units=self.classes,activation=None,flatten=True)
if self.fout:
self.finalout = self.fout()
#self.finalout = self.fout()
pass
def get_node_count(self)->int:
count = 0
count = self.classes
for name in self.op_names:
count += self.__getattribute__(name).count_nodes()
return count
......
......@@ -213,8 +213,8 @@ class Builder:
:returns tuple of two candidate networks the first is of the same size as the previous the the other is depth+1
"""
c0_name = f'can0r{self.round}'
c1_name = f'can1r{self.round}'
c0_name = f'candidate0round{self.round}'
c1_name = f'candidate1round{self.round}'
c0 = CandidateHull(name=c0_name,input=self.input,output=self.output, stack=self.pre_stack,
block_args=self.block_params,batch_size=self.batch_size)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment