CambridgeLandmarks_config.json

{
	"general": {
		"n_freq_print": 5,
		"n_freq_checkpoint": 10,
		"n_workers": 4,
		"device_id": "cuda:0"
	},
	"ems-transposenet":
	{
		"batch_size": 32,
        "equalize_scenes": true,
		"num_scenes": 4,
		"learnable": true,
		"s_x": -3.0,
		"s_q": -6.5,
		"lr": 1e-4,
		"norm": 2,
		"weight_decay": 1e-4,
		"eps": 1e-10,
		"lr_scheduler_gamma": 0.1,
		"lr_scheduler_step_size": 200,
		"n_epochs": 600,
		"num_t_encoder_layers": 6,
		"num_t_decoder_layers": 6,
		"num_rot_encoder_layers": 6,
		"num_rot_decoder_layers": 6,
		"dim_feedforward": 256,
		"hidden_dim":256,
		"dropout": 0.1,
		"nheads": 4,
        "reduction": ["reduction_4", "reduction_3"],
		"freeze": false,
		"freeze_exclude_phrase": ["regressor_head_t", "transformer_t"],
		"no_augment": true,
		"single_scene": false,
		"shallow_mlp": false,
		"apply_positional_encoding": true,
		"num_neighbors": 3,
		"test_optim_iterations": 3,
		"img_size": 64,
		"refine_orientation": null
	},
	"posenet": {
		"batch_size": 32,
		"single_scene": true,
		"num_neighbors": 3,
		"learnable": true,
		"s_x": -3.0,
		"s_q": -6.5,
		"norm": 2,
		"lr": 1e-3,
		"weight_decay": 1e-4,
		"eps": 1e-10,
		"lr_scheduler_step_size": 100,
		"lr_scheduler_gamma": 0.5,
		"n_epochs": 300,
		"shallow_mlp": false,
		"apply_positional_encoding": true,
		"backbone_type": "mobilenet",
		"hidden_dim": 256,
		"test_optim_iterations": 3,
		"num_neighbors": 3,
		"single_scene": true
	}

}