diff options
Diffstat (limited to 'experiments/analyze_oral_a_dynamic_scaling.py')
| -rwxr-xr-x | experiments/analyze_oral_a_dynamic_scaling.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/experiments/analyze_oral_a_dynamic_scaling.py b/experiments/analyze_oral_a_dynamic_scaling.py index 38ce967..972356a 100755 --- a/experiments/analyze_oral_a_dynamic_scaling.py +++ b/experiments/analyze_oral_a_dynamic_scaling.py @@ -102,6 +102,10 @@ def validate_record(record, path, method, depth, seed): and evaluation.get("test_evaluations") == 1 and evaluation.get("test_used_for_selection") is False, f"{path}: evaluation drift") + hardware = record.get("hardware", {}) + require(hardware.get("cuda_device_name") == "NVIDIA GeForce GTX 1080" + and hardware.get("cuda_visible_devices") in ("5", "7"), + f"{path}: unauthorized hardware") epochs = record.get("epochs", []) require(len(epochs) == 200 and all(row.get("epoch") == index + 1 @@ -129,7 +133,7 @@ def validate_record(record, path, method, depth, seed): "elementwise_operations": int(record["work"][ "elementwise_operations_estimate"]), "logical_queries": int(record["work"]["logical_batch_loss_queries"]), - "peak_memory": int(record["hardware"][ + "peak_memory": int(hardware[ "peak_memory_allocated_bytes"]), "source_commit": record["provenance"]["git_commit"], } |