MCPcopy Index your code
hub / github.com/tensorpack/tensorpack / train_with_defaults

Method train_with_defaults

tensorpack/train/base.py:319–345  ·  view source on GitHub ↗

Same as :meth:`train()`, except: 1. Add `extra_callbacks` to callbacks. The default value for `extra_callbacks` is :meth:`DEFAULT_CALLBACKS()`. 2. Default value for `monitors` is :meth:`DEFAULT_MONITORS()`. 3. Provide default values for every option excep

(
            self, _sentinel=None,
            callbacks=None, monitors=None,
            session_creator=None, session_init=None,
            steps_per_epoch=None, starting_epoch=1, max_epoch=9999999,
            extra_callbacks=None)

Source from the content-addressed store, hash-verified

317 self.main_loop(steps_per_epoch, starting_epoch, max_epoch)
318
319 def train_with_defaults(
320 self, _sentinel=None,
321 callbacks=None, monitors=None,
322 session_creator=None, session_init=None,
323 steps_per_epoch=None, starting_epoch=1, max_epoch=9999999,
324 extra_callbacks=None):
325 """
326 Same as :meth:`train()`, except:
327
328 1. Add `extra_callbacks` to callbacks. The default value for
329 `extra_callbacks` is :meth:`DEFAULT_CALLBACKS()`.
330 2. Default value for `monitors` is :meth:`DEFAULT_MONITORS()`.
331 3. Provide default values for every option except `steps_per_epoch`.
332 """
333 assert _sentinel is None, "Please call `train_with_defaults` with keyword arguments only!"
334 callbacks = copy.copy(callbacks or [])
335 monitors = DEFAULT_MONITORS() if monitors is None else monitors
336 extra_callbacks = DEFAULT_CALLBACKS() if extra_callbacks is None else extra_callbacks
337 callbacks.extend(extra_callbacks)
338
339 assert steps_per_epoch is not None
340 session_creator = session_creator or NewSessionCreator()
341 session_init = session_init or JustCurrentSession()
342
343 self.train(callbacks, monitors,
344 session_creator, session_init,
345 steps_per_epoch, starting_epoch, max_epoch)
346
347 def __new__(cls, *args, **kwargs):
348 if (len(args) > 0 and isinstance(args[0], TrainConfig)) \

Callers 13

fitMethod · 0.80
launch_train_with_configFunction · 0.80
DiscoGAN-CelebA.pyFile · 0.80
WGAN.pyFile · 0.80
Image2Image.pyFile · 0.80
CycleGAN.pyFile · 0.80
BEGAN.pyFile · 0.80
Improved-WGAN.pyFile · 0.80
InfoGAN-mnist.pyFile · 0.80
DCGAN.pyFile · 0.80

Calls 5

trainMethod · 0.95
DEFAULT_MONITORSFunction · 0.85
DEFAULT_CALLBACKSFunction · 0.85
NewSessionCreatorClass · 0.85
JustCurrentSessionClass · 0.85

Tested by 1