Skip to content
Snippets Groups Projects
Verified Commit 9355a3aa authored by Björn Ludwig's avatar Björn Ludwig
Browse files

refactor(solve_instances_in_parallel): utilize new timing module

parent 230cfe4f
Branches
No related tags found
No related merge requests found
...@@ -17,6 +17,7 @@ from lp_nn_robustness_verification.data_acquisition.uncertain_inputs import ( ...@@ -17,6 +17,7 @@ from lp_nn_robustness_verification.data_acquisition.uncertain_inputs import (
from lp_nn_robustness_verification.data_types import UncertainArray from lp_nn_robustness_verification.data_types import UncertainArray
from lp_nn_robustness_verification.linear_program import RobustVerifier from lp_nn_robustness_verification.linear_program import RobustVerifier
from lp_nn_robustness_verification.pre_processing import LinearInclusion from lp_nn_robustness_verification.pre_processing import LinearInclusion
from lp_nn_robustness_verification.timing import write_current_timing_stats
def solve_and_store_timed_solutions(task_id: int) -> None: def solve_and_store_timed_solutions(task_id: int) -> None:
...@@ -69,26 +70,15 @@ def solve_and_store_timed_solutions(task_id: int) -> None: ...@@ -69,26 +70,15 @@ def solve_and_store_timed_solutions(task_id: int) -> None:
optimization = RobustVerifier(linear_inclusion) optimization = RobustVerifier(linear_inclusion)
optimization.solve() optimization.solve()
yappi.stop() yappi.stop()
write_current_timing_stats(
f"{size_scaler * 11}_inputs_and_{depth}_layers_with_sample_"
f"{idx_start}_and_seed_{seed}_"
f"timings.txt",
"Everything has been done",
"a",
)
if optimization.model.getSols(): if optimization.model.getSols():
solved = True solved = True
with open(
f"{size_scaler * 11}_inputs_and_{depth}_layers_with_sample_"
f"{idx_start}_and_seed_{seed}_"
f"timings.txt",
"a",
encoding="utf-8",
) as timings_file:
timings_file.write(
f"\n==========================================================="
f"===========================\n"
f"Timings for {size_scaler * 11} inputs and {depth} "
f"{'layers' if depth > 1 else 'layer'} with sample 0 and seed 0"
f"\n==========================================================="
f"===========================\n"
)
yappi.get_func_stats().print_all(
out=timings_file, columns={0: ("name", 180), 3: ("ttot", 8)}
)
optimization.model.writeProblem( optimization.model.writeProblem(
filename=( filename=(
f"{size_scaler * 11}_inputs_and_{depth}_layers_with_sample_" f"{size_scaler * 11}_inputs_and_{depth}_layers_with_sample_"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment