Searched refs:seg_trainer (Results 1 – 2 of 2) sorted by relevance
365 dnn_trainer<seg_bnet_type> seg_trainer(seg_net, sgd(weight_decay, momentum)); in train_segmentation_network() local366 seg_trainer.be_verbose(); in train_segmentation_network()367 seg_trainer.set_learning_rate(initial_learning_rate); in train_segmentation_network()368 seg_trainer.set_synchronization_file(synchronization_file_name, std::chrono::minutes(10)); in train_segmentation_network()369 seg_trainer.set_iterations_without_progress_threshold(2000); in train_segmentation_network()373 cout << seg_trainer << endl; in train_segmentation_network()470 while (seg_trainer.get_learning_rate() >= 1e-4) in train_segmentation_network()485 seg_trainer.train_one_step(samples, labels); in train_segmentation_network()499 seg_trainer.get_net(); in train_segmentation_network()