(lua, pth, maps)
| 9 | |
| 10 | |
| 11 | def weight_assign(lua, pth, maps): |
| 12 | for k, v in maps.items(): |
| 13 | getattr(pth, k).weight = nn.Parameter(lua.get(v).weight.float()) |
| 14 | getattr(pth, k).bias = nn.Parameter(lua.get(v).bias.float()) |
| 15 | |
| 16 | |
| 17 | def photo_wct_loader(p_wct): |