Config

The global DeepOBS config.

deepobs.config.get_testproblem_default_setting(testproblem)

Returns default settings for the batch_size and the num_epochs for testproblem (if available).

Parameters:testproblem (str) -- Test problem for which to return the default setting. Returns:A dictionary with the default values for batch_size and num_epochs Return type:dict

deepobs.config.set_framework(framework)

Sets the current used framework. This is relevant for the higher level Tuner module of DeepOBS.

Parameters:framework (str) -- Can be 'pytorch' or 'tensorflow'

deepobs.config.set_data_dir(data_dir)

Sets the data directory.

Parameters:data_dir (str) -- Path to the data folder.