| {"classification": true, "verbose": false, "model_name": "sa_w_1.0_fpn_p_384_all_trans_all_data_clean_35_ep", "data_path": "/services/scratch/perception/lairale/nbm/faster_rcnn", "save_dir": "./models", "pretrain_conv_net": true, "backbone": "vgg", "conv_net_path": "checkpoints/vgg16_bn-6c64b313.pth", "fpn_rpn": false, "fpn": true, "fpn_p_channels": 384, "fpn_o_channels": 256, "use_biophonia": true, "img_width": 1024, "normalize_input": true, "noise_strength": 0, "self_attention": true, "encode_frequency": true, "position_encoding": false, "transform": true, "scales_factor_low": 0, "scales_factor_high": 4, "rpn_neg_label": 0.3, "rpn_pos_label": 0.7, "rpn_batchsize": 16, "rpn_fg_fraction": 0.5, "pre_nms_topN": 3000, "min_threshold": 5, "nms_thresh": 0.7, "post_nms_topN": 1000, "post_nms_topN_eval": 50, "pre_nms_topN_eval": 500, "rcnn_batch_size": 16, "rcnn_fg_prop": 0.4, "fg_threshold": 0.5, "bg_threshold_lo": 0.1, "bg_threshold_hi": 0.5, "roi_pool_h": 2, "roi_pool_w": 2, "hidden_size": 4096, "top_pyramid_roi_size": 128, "rcnn_attention": false, "lambda_reg_rpn_loss": 1.0, "lambda_reg_rcnn_loss": 1.0, "learning_rate": 0.0001, "n_epochs": 35, "batch_size": 8, "val_size": 20, "validation_prop": 0.005, "save_every": 50, "scheduler_gamma": 0.1, "cv_idx": 4} |