Method trainable_modules

models/training.py:193–195 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

191
192
193	def trainable_modules(self):
194	trainable_modules = filter(lambda p: p.requires_grad, self.parameters())
195	return trainable_modules
196
197
198	def to(self, args, *kwargs):

launch_training_taskFunction · 0.80

no outgoing calls

no test coverage detected