loss_history_train = {}
loss_history_eval = {}
-
+
+ # --------------------------------------------------------------
+
# TensorboardX
if not args['__raytune_running__'] and param['tensorboard']:
from tensorboardX import SummaryWriter
@@ -1055,7 +1057,7 @@ Source code for icenet.deep.iceboost
# ---------------------------------------------------------
# Choose weight mode
if np.min(w_trn) < 0.0 or np.min(w_val) < 0.0:
- print(f'Negative weights in the sample -- handled via custom loss')
+ print(f'Negative weights in the sample -- handled via custom loss', 'magenta')
out_weights_on = True
if not use_custom:
diff --git a/_modules/icenet/deep/losstools.html b/_modules/icenet/deep/losstools.html
index cadc6530..03a0bbaf 100644
--- a/_modules/icenet/deep/losstools.html
+++ b/_modules/icenet/deep/losstools.html
@@ -32,7 +32,7 @@
- 0.1.2.4
+ 0.1.2.5