Skip to content

Instantly share code, notes, and snippets.

@ganler
Created October 14, 2022 20:46
Show Gist options
  • Save ganler/033f3b2c56ec02c4c545c337f7a478b2 to your computer and use it in GitHub Desktop.
Save ganler/033f3b2c56ec02c4c545c337f7a478b2 to your computer and use it in GitHub Desktop.
Quickly fix grad experiment bugs in NNSmith artifact version.
diff --git a/experiments/input_search.py b/experiments/input_search.py
index 3a2eec0..f82f55e 100644
--- a/experiments/input_search.py
+++ b/experiments/input_search.py
@@ -50,7 +50,7 @@ def mknets(args, exp_seed):
os.makedirs(model_path)
__DIFF_CACHE__ = 'config/diff.pkl'
differentiable_ops = rewrite_op_dtype(
- ALL_OP_TYPES, factory=None, diff=True, verbose=True, cache=__DIFF_CACHE__)
+ ALL_OP_TYPES, factory=None, diff=True, verbose=False, cache=__DIFF_CACHE__)
print(differentiable_ops)
results = {
'model_seed': [],
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment