(self, path, src_layers)
| 50 | }) |
| 51 | |
| 52 | def load(self, path, src_layers): |
| 53 | self.src_layers = src_layers |
| 54 | walker = weights_walker(path) |
| 55 | |
| 56 | for i, layer in enumerate(src_layers): |
| 57 | if layer.type not in self.VAR_LAYER: continue |
| 58 | self.src_key.append([layer]) |
| 59 | |
| 60 | if walker.eof: new = None |
| 61 | else: |
| 62 | args = layer.signature |
| 63 | new = dark.darknet.create_darkop(*args) |
| 64 | self.vals.append(new) |
| 65 | |
| 66 | if new is None: continue |
| 67 | order = self._W_ORDER[new.type] |
| 68 | for par in order: |
| 69 | if par not in new.wshape: continue |
| 70 | val = walker.walk(new.wsize[par]) |
| 71 | new.w[par] = val |
| 72 | new.finalize(walker.transpose) |
| 73 | |
| 74 | if walker.path is not None: |
| 75 | assert walker.offset == walker.size, \ |
| 76 | 'expect {} bytes, found {}'.format( |
| 77 | walker.offset, walker.size) |
| 78 | print('Successfully identified {} bytes'.format( |
| 79 | walker.offset)) |
| 80 | |
| 81 | class checkpoint_loader(loader): |
| 82 | """ |
no test coverage detected