- added helper function for fire

Todos: matching parameters (thresholds, trails etc. need estimation)
process output information
parent 7223542d
......@@ -470,11 +470,20 @@ def check_experiment_wide_parameter(parameter, parameter_name, required_type, al
return parameter
torch.backends.cudnn.benchmark = True
def fire_experiment(experiment_name, stages, dataset, validation_split=0.2, batch_size=8, num_workers=4, epochs=30,
learning_rate=0.008, learning_rate_gamma=0.5, learning_rate_step_size=10, weight_decay=0.0001,
load_from=None, target_layer="layer3", use_l2_normalization=True, pca_dataset=None, pca_load=None):
ExperimentManager(experiment_name, stages, dataset, validation_split, batch_size, num_workers, epochs,
learning_rate, learning_rate_gamma, learning_rate_step_size, weight_decay, load_from,
target_layer, use_l2_normalization, pca_dataset, pca_load)
#torch.backends.cudnn.benchmark = True
#exp = ExperimentManager("30_epoch_run", {"finetuning","keypoints","retrieval"}, {"finetuning": "../Datasets/Landmarks", "keypoints": "../../Datasets/Landmarks", "retrieval": "../../Datasets/Oxford"}, epochs=30)
#exp = ExperimentManager("30_epoch_run", {"retrieval"}, {"retrieval": "../Datasets/Oxford"}, load_from={"retrieval":"Experiments/30_epoch_run/keypoints/e3f41a23_d0f3a26d.pth"})
#exp = ExperimentManager("layer4", {"retrieval"}, {"retrieval": "../Datasets/Oxford"}, epochs=1, target_layer='layer4', load_from={"retrieval": "Experiments/layer4/keypoints/90a267db_05ef12cd.pth"})
#exp = ExperimentManager("all_stages", {"keypoints","retrieval"}, {"keypoints":"../Datasets/Landmarks", "retrieval": "../Datasets/Oxford"}, epochs=1, load_from={"keypoints":"Experiments/all_stages/keypoints/c48957fe_6d39b7dc.pth"})
#exp = ExperimentManager("variable target layer", {"keypoints"}, "../Datasets/Landmarks", epochs=1, load_from={"keypoints":"Experiments/variable target layer/finetuning/5db43e8d.pth"})
if __name__ == '__main__':
fire.Fire(ExperimentManager)
fire.Fire(fire_experiment)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment