{ "n_dim": 2, "axes": "YXC", "n_channel_in": 3, "n_channel_out": 33, "train_checkpoint": "weights_best.h5", "train_checkpoint_last": "weights_last.h5", "train_checkpoint_epoch": "weights_now.h5", "n_rays": 32, "grid": [ 2, 2 ], "backbone": "unet", "unet_n_depth": 3, "unet_kernel_size": [ 3, 3 ], "unet_n_filter_base": 32, "unet_n_conv_per_depth": 2, "unet_pool": [ 2, 2 ], "unet_activation": "relu", "unet_last_activation": "relu", "unet_batch_norm": false, "unet_dropout": 0.0, "unet_prefix": "", "net_conv_after_unet": 128, "net_input_shape": [ null, null, 3 ], "net_mask_shape": [ null, null, 1 ], "train_shape_completion": false, "train_completion_crop": 32, "train_patch_size": [ 512, 512 ], "train_background_reg": 0.0001, "train_foreground_only": 0.9, "train_dist_loss": "mae", "train_loss_weights": [ 1, 0.1 ], "train_epochs": 200, "train_steps_per_epoch": 50, "train_learning_rate": 0.0003, "train_batch_size": 8, "train_n_val_patches": 3, "train_tensorboard": true, "train_reduce_lr": { "factor": 0.5, "patience": 20, "min_delta": 0 }, "use_gpu": true }