1. 25 Jun, 2020 2 commits
    • Ignacio Pickering's avatar
      Tests cleanup (#495) · 28ecce79
      Ignacio Pickering authored
      * Remove test gradient from aev tests
      
      * Add a dedicated test gradient for one model, and make it feasible
      
      * add test_grad to unittests
      
      * Add some comments and fix formatting
      
      * flake8
      
      * flake8
      
      * remove unused var
      
      * delete transform function, which was legacy from old pytorch probably
      
      * Delete random_skip function, which was unused
      
      * Allow testNeuroChem to use cuda if available, which makes it much faster
      
      * small modification for consistency
      
      * remove unused import
      
      * simplify device call
      
      * remove unused variable
      
      * add main to test_grad
      28ecce79
    • Ignacio Pickering's avatar
      Fix gradcheck test (#493) · 9a03bc62
      Ignacio Pickering authored
      * Remove test gradient from aev tests
      
      * Add a dedicated test gradient for one model, and make it feasible
      
      * add test_grad to unittests
      
      * Add some comments and fix formatting
      
      * flake8
      
      * flake8
      
      * remove unused var
      9a03bc62