소스 검색

Update pytorch.js

Lutz Roeder 4 년 전
부모
커밋
1e6dda0316
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      source/pytorch.js

+ 1 - 1
source/pytorch.js

@@ -3350,7 +3350,7 @@ pytorch.Utility = class {
                 for (const key of Object.keys(obj)) {
                     const value = obj[key];
                     if (key.indexOf('optim') !== -1 || key.indexOf('opt') !== -1) {
-                        if (value.state && value.param_groups) {
+                        if (value === null || (value.state && value.param_groups)) {
                             continue;
                         }
                     }