Method multi_gpu_wrapper

train_generation.py:465–466 · view source on GitHub ↗

(self, f)

Source from the content-addressed store, hash-verified

463	self.model.eval()
464
465	def multi_gpu_wrapper(self, f):
466	self.model = f(self.model)
467
468
469	def get_betas(schedule_type, b_start, b_end, time_num):

trainFunction · 0.95

no outgoing calls

no test coverage detected