Merge pull request #8 from BrandoZhang/patch-1

Fix: upfirdn2d_plugin compilation issue.
universome · May 30, 2021 · 4607b11 · 4607b11
2 parents 9699eba + 246e2f2
commit 4607b11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/main.yml b/configs/main.yml
@@ -2,7 +2,7 @@ env:
   python_bin: "python" # Path to the python bin
   base_project_dir: ${hydra:runtime.cwd}
   before_train_commands: []
-  torch_extensions_dir: "/tmp/torch_extensions"
+  torch_extensions_dir: "~/.cache/torch_extensions"
   data_dir: "data"
   objects_to_copy:
     - ${env.base_project_dir}/configs