def chamfer_loss_logger(current_epoch, epochs, steps_per_epoch): bar = Bar( "Epoch %d/%d" % (current_epoch, epochs), suffix=( "%(index)d/%(max)d - loss: %(loss).7f - " "pcl_to_prim: %(pcl_to_prim_loss).7f - " "prim_to_pcl: %(prim_to_pcl_loss).7f - " # "bern_reg: %(bernoulli_regularizer).8f " # "entr_bern_reg: %(entropy_bernoulli_regularizer).8f " # "sp_reg: %(sparsity_regularizer).8f " # "overl_reg: %(overlapping_regularizer).8f " # "parsimony_reg: %(parsimony_regularizer).8f " "exp_n_prims: %(exp_n_prims).4f" ""), max=steps_per_epoch, loss=0.0, pcl_to_prim_loss=0.0, prim_to_pcl_loss=0.0, sparsity_regularizer=0.0, entropy_bernoulli_regularizer=0.0, bernoulli_regularizer=0.0, parsimony_regularizer=0.0, overlapping_regularizer=0.0, exp_n_prims=0.0) bar.hide_cursor = False return bar
def matrix_loss_logger(current_epoch, epochs, steps_per_epoch): bar = Bar( "Epoch %d/%d" % (current_epoch, epochs), suffix=("%(index)d/%(max)d - loss: %(loss).7f - " ""), max=steps_per_epoch, loss=0.0, ) bar.hide_cursor = False return bar