Fix bugs in .to(cpu)

2021-03-30 12:25:47 +00:00
parent e5ec43e04a
commit 756218974f
4 changed files with 17 additions and 1 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -132,3 +132,4 @@ mlruns*
 outputs

 pytest_cache
+*.pkl