Method get_init_args

tasks/AutoMolecule3D/code/visnet/priors.py:76–77 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

74	self.atomref.weight.data.copy_(self.initial_atomref)
75
76	def get_init_args(self):
77	return dict(max_z=self.initial_atomref.size(0))
78
79	def forward(self, x, z):
80	return x + self.atomref(z)

mainFunction · 0.45

no outgoing calls

no test coverage detected