tensorcv.train package

Submodules

tensorcv.train.base module

class tensorcv.train.base.Trainer(config)

Bases: object

base class for trainer

epochs_completed

get_global_step

main_loop()

register_callback(cb)

register_monitor(monitor)

setup()

setup_graph()

train()

tensorcv.train.config module

class tensorcv.train.config.TrainConfig(dataflow=None, model=None, callbacks=[], session_creator=None, monitors=None, batch_size=1, max_epoch=100, summary_periodic=None, is_load=False, model_name=None, default_dirs=None)

Bases: object

callbacks

class tensorcv.train.config.GANTrainConfig(dataflow=None, model=None, discriminator_callbacks=[], generator_callbacks=[], session_creator=None, monitors=None, batch_size=1, max_epoch=100, summary_d_periodic=None, summary_g_periodic=None, default_dirs=None)

Bases: tensorcv.train.config.TrainConfig

dis_callbacks

gen_callbacks

tensorcv.train.simple module

class tensorcv.train.simple.SimpleFeedTrainer(config)

Bases: tensorcv.train.base.Trainer

single optimizer

Module contents