Method multi_gpu_wrapper

test_generation.py:335–336 · view source on GitHub ↗

(self, f)

Source from the content-addressed store, hash-verified

333	self.model.eval()
334
335	def multi_gpu_wrapper(self, f):
336	self.model = f(self.model)
337
338
339	def get_betas(schedule_type, b_start, b_end, time_num):

mainFunction · 0.95

no outgoing calls

no test coverage detected