| | |
|---|
| | | oss_bucket=None, |
|---|
| | | scope_map=[], |
|---|
| | | excludes=None, |
|---|
| | | ignore_mismatch=False, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | """Load a model state and set it to the model. |
|---|
| | |
|---|
| | | |
|---|
| | | if k_src in src_state.keys(): |
|---|
| | | if ignore_init_mismatch and dst_state[k].shape != src_state[k_src].shape: |
|---|
| | | print(f"ignore_mismatch:{ignore_mismatch}, dst: {k, dst_state[k].shape}, src: {k_src, src_state[k_src].shape}") |
|---|
| | | print(f"ignore_init_mismatch:{ignore_init_mismatch}, dst: {k, dst_state[k].shape}, src: {k_src, src_state[k_src].shape}") |
|---|
| | | else: |
|---|
| | | dst_state[k] = src_state[k_src] |
|---|
| | | |
|---|
