MCPcopy Create free account
hub / github.com/albertpumarola/GANimation / set_train

Method set_train

models/ganimation.py:110–113  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

108 self._input_desired_cond = self._input_desired_cond.cuda(self._gpu_ids[0], async=True)
109
110 def set_train(self):
111 self._G.train()
112 self._D.train()
113 self._is_train = True
114
115 def set_eval(self):
116 self._G.eval()

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected