search

xinntao / Real-ESRGAN

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.
BSD 3-Clause "New" or "Revised" License
28.63k stars 3.59k forks source link

RuntimeError: Error(s) in loading state_dict #707

Open ajeema opened 1 year ago

ajeema commented 1 year ago

I have trained the net and am trying to now train the gan. The net worked and finished and the gan works until interrupted (Colab). When I try and resume net training it works well. When trying to resume gan training from last saved state I get an error complaining about missing and unexpected key. See below

Missing key(s) in state_dict: "conv_first.weight", "conv_first.bias", "body.0.srdb1.conv1x1.weight", "body.0.srdb1.conv1.weight", "body.0.srdb1.conv1.bias", "body.0.srdb1.conv2.weight", "body.0.srdb1.conv2.bias", "body.0.srdb1.conv3.weight", "body.0.srdb1.conv3.bias", "body.0.srdb1.conv4.weight", "body.0.srdb1.conv4.bias", "body.0.srdb1.conv5.weight", "body.0.srdb1.conv5.bias", "body.0.srdb2.conv1x1.weight", "body.0.srdb2.conv1.weight", "body.0.srdb2.conv1.bias", "body.0.srdb2.conv2.weight", "body.0.srdb2.conv2.bias", "body.0.srdb2.conv3.weight", "body.0.srdb2.conv3.bias", "body.0.srdb2.conv4.weight", "body.0.srdb2.conv4.bias", "body.0.srdb2.conv5.weight", "body.0.srdb2.conv5.bias", "body.0.srdb3.conv1x1.weight", "body.0.srdb3.conv1.weight", "body.0.srdb3.conv1.bias", "body.0.srdb3.conv2.weight", "body.0.srdb3.conv2.bias", "body.0.srdb3.conv3.weight", "body.0.srdb3.conv3.bias", "body.0.srdb3.conv4.weight", "body.0.srdb3.conv4.bias", "body.0.srdb3.conv5.weight", "body.0.srdb3.conv5.bias", "body.1.srdb1.conv1x1.weight", "body.1.srdb1.conv1.weight", "body.1.srdb1.conv1.bias", "body.1.srdb1.conv2.weight", "body.1.srdb1.conv2.bias", "body.1.srdb1.conv3.weight", "body.1.srdb1.conv3.bias", "body.1.srdb1.conv4.weight", "body.1.srdb1.conv4.bias", "body.1.srdb1.conv5.weight", "body.1.srdb1.conv5.bias", "body.1.srdb2.conv1x1.weight", "body.1.srdb2.conv1.weight", "body.1.srdb2.conv1.bias", "body.1.srdb2.conv2.weight", "body.1.srdb2.conv2.bias", "body.1.srdb2.conv3.weight", "body.1.srdb2.conv3.bias", "body.1.srdb2.conv4.weight", "body.1.srdb2.conv4.bias", "body.1.srdb2.conv5.weight", "body.1.srdb2.conv5.bias", "body.1.srdb3.conv1x1.weight", "body.1.srdb3.conv1.weight", "body.1.srdb3.conv1.bias", "body.1.srdb3.conv2.weight", "body.1.srdb3.conv2.bias", "body.1.srdb3.conv3.weight", "body.1.srdb3.conv3.bias", "body.1.srdb3.conv4.weight", "body.1.srdb3.conv4.bias", "body.1.srdb3.conv5.weight", "body.1.srdb3.conv5.bias", "body.2.srdb1.conv1x1.weight", "body.2.srdb1.conv1.weight", "body.2.srdb1.conv1.bias", "body.2.srdb1.conv2.weight", "body.2.srdb1.conv2.bias", "body.2.srdb1.conv3.weight", "body.2.srdb1.conv3.bias", "body.2.srdb1.conv4.weight", "body.2.srdb1.conv4.bias", "body.2.srdb1.conv5.weight", "body.2.srdb1.conv5.bias", "body.2.srdb2.conv1x1.weight", "body.2.srdb2.conv1.weight", "body.2.srdb2.conv1.bias", "body.2.srdb2.conv2.weight", "body.2.srdb2.conv2.bias", "body.2.srdb2.conv3.weight", "body.2.srdb2.conv3.bias", "body.2.srdb2.conv4.weight", "body.2.srdb2.conv4.bias", "body.2.srdb2.conv5.weight", "body.2.srdb2.conv5.bias", "body.2.srdb3.conv1x1.weight", "body.2.srdb3.conv1.weight", "body.2.srdb3.conv1.bias", "body.2.srdb3.conv2.weight", "body.2.srdb3.conv2.bias", "body.2.srdb3.conv3.weight", "body.2.srdb3.conv3.bias", "body.2.srdb3.conv4.weight", "body.2.srdb3.conv4.bias", "body.2.srdb3.conv5.weight", "body.2.srdb3.conv5.bias", "body.3.srdb1.conv1x1.weight", "body.3.srdb1.conv1.weight", "body.3.srdb1.conv1.bias", "body.3.srdb1.conv2.weight", "body.3.srdb1.conv2.bias", "body.3.srdb1.conv3.weight", "body.3.srdb1.conv3.bias", "body.3.srdb1.conv4.weight", "body.3.srdb1.conv4.bias", "body.3.srdb1.conv5.weight", "body.3.srdb1.conv5.bias", "body.3.srdb2.conv1x1.weight", "body.3.srdb2.conv1.weight", "body.3.srdb2.conv1.bias", "body.3.srdb2.conv2.weight", "body.3.srdb2.conv2.bias", "body.3.srdb2.conv3.weight", "body.3.srdb2.conv3.bias", "body.3.srdb2.conv4.weight", "body.3.srdb2.conv4.bias", "body.3.srdb2.conv5.weight", "body.3.srdb2.conv5.bias", "body.3.srdb3.conv1x1.weight", "body.3.srdb3.conv1.weight", "body.3.srdb3.conv1.bias", "body.3.srdb3.conv2.weight", "body.3.srdb3.conv2.bias", "body.3.srdb3.conv3.weight", "body.3.srdb3.conv3.bias", "body.3.srdb3.conv4.weight", "body.3.srdb3.conv4.bias", "body.3.srdb3.conv5.weight", "body.3.srdb3.conv5.bias", "body.4.srdb1.conv1x1.weight", "body.4.srdb1.conv1.weight", "body.4.srdb1.conv1.bias", "body.4.srdb1.conv2.weight", "body.4.srdb1.conv2.bias", "body.4.srdb1.conv3.weight", "body.4.srdb1.conv3.bias", "body.4.srdb1.conv4.weight", "body.4.srdb1.conv4.bias", "body.4.srdb1.conv5.weight", "body.4.srdb1.conv5.bias", "body.4.srdb2.conv1x1.weight", "body.4.srdb2.conv1.weight", "body.4.srdb2.conv1.bias", "body.4.srdb2.conv2.weight", "body.4.srdb2.conv2.bias", "body.4.srdb2.conv3.weight", "body.4.srdb2.conv3.bias", "body.4.srdb2.conv4.weight", "body.4.srdb2.conv4.bias", "body.4.srdb2.conv5.weight", "body.4.srdb2.conv5.bias", "body.4.srdb3.conv1x1.weight", "body.4.srdb3.conv1.weight", "body.4.srdb3.conv1.bias", "body.4.srdb3.conv2.weight", "body.4.srdb3.conv2.bias", "body.4.srdb3.conv3.weight", "body.4.srdb3.conv3.bias", "body.4.srdb3.conv4.weight", "body.4.srdb3.conv4.bias", "body.4.srdb3.conv5.weight", "body.4.srdb3.conv5.bias", "body.5.srdb1.conv1x1.weight", "body.5.srdb1.conv1.weight", "body.5.srdb1.conv1.bias", "body.5.srdb1.conv2.weight", "body.5.srdb1.conv2.bias", "body.5.srdb1.conv3.weight", "body.5.srdb1.conv3.bias", "body.5.srdb1.conv4.weight", "body.5.srdb1.conv4.bias", "body.5.srdb1.conv5.weight", "body.5.srdb1.conv5.bias", "body.5.srdb2.conv1x1.weight", "body.5.srdb2.conv1.weight", "body.5.srdb2.conv1.bias", "body.5.srdb2.conv2.weight", "body.5.srdb2.conv2.bias", "body.5.srdb2.conv3.weight", "body.5.srdb2.conv3.bias", "body.5.srdb2.conv4.weight", "body.5.srdb2.conv4.bias", "body.5.srdb2.conv5.weight", "body.5.srdb2.conv5.bias", "body.5.srdb3.conv1x1.weight", "body.5.srdb3.conv1.weight", "body.5.srdb3.conv1.bias", "body.5.srdb3.conv2.weight", "body.5.srdb3.conv2.bias", "body.5.srdb3.conv3.weight", "body.5.srdb3.conv3.bias", "body.5.srdb3.conv4.weight", "body.5.srdb3.conv4.bias", "body.5.srdb3.conv5.weight", "body.5.srdb3.conv5.bias", "body.6.srdb1.conv1x1.weight", "body.6.srdb1.conv1.weight", "body.6.srdb1.conv1.bias", "body.6.srdb1.conv2.weight", "body.6.srdb1.conv2.bias", "body.6.srdb1.conv3.weight", "body.6.srdb1.conv3.bias", "body.6.srdb1.conv4.weight", "body.6.srdb1.conv4.bias", "body.6.srdb1.conv5.weight", "body.6.srdb1.conv5.bias", "body.6.srdb2.conv1x1.weight", "body.6.srdb2.conv1.weight", "body.6.srdb2.conv1.bias", "body.6.srdb2.conv2.weight", "body.6.srdb2.conv2.bias", "body.6.srdb2.conv3.weight", "body.6.srdb2.conv3.bias", "body.6.srdb2.conv4.weight", "body.6.srdb2.conv4.bias", "body.6.srdb2.conv5.weight", "body.6.srdb2.conv5.bias", "body.6.srdb3.conv1x1.weight", "body.6.srdb3.conv1.weight", "body.6.srdb3.conv1.bias", "body.6.srdb3.conv2.weight", "body.6.srdb3.conv2.bias", "body.6.srdb3.conv3.weight", "body.6.srdb3.conv3.bias", "body.6.srdb3.conv4.weight", "body.6.srdb3.conv4.bias", "body.6.srdb3.conv5.weight", "body.6.srdb3.conv5.bias", "body.7.srdb1.conv1x1.weight", "body.7.srdb1.conv1.weight", "body.7.srdb1.conv1.bias", "body.7.srdb1.conv2.weight", "body.7.srdb1.conv2.bias", "body.7.srdb1.conv3.weight", "body.7.srdb1.conv3.bias", "body.7.srdb1.conv4.weight", "body.7.srdb1.conv4.bias", "body.7.srdb1.conv5.weight", "body.7.srdb1.conv5.bias", "body.7.srdb2.conv1x1.weight", "body.7.srdb2.conv1.weight", "body.7.srdb2.conv1.bias", "body.7.srdb2.conv2.weight", "body.7.srdb2.conv2.bias", "body.7.srdb2.conv3.weight", "body.7.srdb2.conv3.bias", "body.7.srdb2.conv4.weight", "body.7.srdb2.conv4.bias", "body.7.srdb2.conv5.weight", "body.7.srdb2.conv5.bias", "body.7.srdb3.conv1x1.weight", "body.7.srdb3.conv1.weight", "body.7.srdb3.conv1.bias", "body.7.srdb3.conv2.weight", "body.7.srdb3.conv2.bias", "body.7.srdb3.conv3.weight", "body.7.srdb3.conv3.bias", "body.7.srdb3.conv4.weight", "body.7.srdb3.conv4.bias", "body.7.srdb3.conv5.weight", "body.7.srdb3.conv5.bias", "body.8.srdb1.conv1x1.weight", "body.8.srdb1.conv1.weight", "body.8.srdb1.conv1.bias", "body.8.srdb1.conv2.weight", "body.8.srdb1.conv2.bias", "body.8.srdb1.conv3.weight", "body.8.srdb1.conv3.bias", "body.8.srdb1.conv4.weight", "body.8.srdb1.conv4.bias", "body.8.srdb1.conv5.weight", "body.8.srdb1.conv5.bias", "body.8.srdb2.conv1x1.weight", "body.8.srdb2.conv1.weight", "body.8.srdb2.conv1.bias", "body.8.srdb2.conv2.weight", "body.8.srdb2.conv2.bias", "body.8.srdb2.conv3.weight", "body.8.srdb2.conv3.bias", "body.8.srdb2.conv4.weight", "body.8.srdb2.conv4.bias", "body.8.srdb2.conv5.weight", "body.8.srdb2.conv5.bias", "body.8.srdb3.conv1x1.weight", "body.8.srdb3.conv1.weight", "body.8.srdb3.conv1.bias", "body.8.srdb3.conv2.weight", "body.8.srdb3.conv2.bias", "body.8.srdb3.conv3.weight", "body.8.srdb3.conv3.bias", "body.8.srdb3.conv4.weight", "body.8.srdb3.conv4.bias", "body.8.srdb3.conv5.weight", "body.8.srdb3.conv5.bias", "body.9.srdb1.conv1x1.weight", "body.9.srdb1.conv1.weight", "body.9.srdb1.conv1.bias", "body.9.srdb1.conv2.weight", "body.9.srdb1.conv2.bias", "body.9.srdb1.conv3.weight", "body.9.srdb1.conv3.bias", "body.9.srdb1.conv4.weight", "body.9.srdb1.conv4.bias", "body.9.srdb1.conv5.weight", "body.9.srdb1.conv5.bias", "body.9.srdb2.conv1x1.weight", "body.9.srdb2.conv1.weight", "body.9.srdb2.conv1.bias", "body.9.srdb2.conv2.weight", "body.9.srdb2.conv2.bias", "body.9.srdb2.conv3.weight", "body.9.srdb2.conv3.bias", "body.9.srdb2.conv4.weight", "body.9.srdb2.conv4.bias", "body.9.srdb2.conv5.weight", "body.9.srdb2.conv5.bias", "body.9.srdb3.conv1x1.weight", "body.9.srdb3.conv1.weight", "body.9.srdb3.conv1.bias", "body.9.srdb3.conv2.weight", "body.9.srdb3.conv2.bias", "body.9.srdb3.conv3.weight", "body.9.srdb3.conv3.bias", "body.9.srdb3.conv4.weight", "body.9.srdb3.conv4.bias", "body.9.srdb3.conv5.weight", "body.9.srdb3.conv5.bias", "body.10.srdb1.conv1x1.weight", "body.10.srdb1.conv1.weight", "body.10.srdb1.conv1.bias", "body.10.srdb1.conv2.weight", "body.10.srdb1.conv2.bias", "body.10.srdb1.conv3.weight", "body.10.srdb1.conv3.bias", "body.10.srdb1.conv4.weight", "body.10.srdb1.conv4.bias", "body.10.srdb1.conv5.weight", "body.10.srdb1.conv5.bias", "body.10.srdb2.conv1x1.weight", "body.10.srdb2.conv1.weight", "body.10.srdb2.conv1.bias", "body.10.srdb2.conv2.weight", "body.10.srdb2.conv2.bias", "body.10.srdb2.conv3.weight", "body.10.srdb2.conv3.bias", "body.10.srdb2.conv4.weight", "body.10.srdb2.conv4.bias", "body.10.srdb2.conv5.weight", "body.10.srdb2.conv5.bias", "body.10.srdb3.conv1x1.weight", "body.10.srdb3.conv1.weight", "body.10.srdb3.conv1.bias", "body.10.srdb3.conv2.weight", "body.10.srdb3.conv2.bias", "body.10.srdb3.conv3.weight", "body.10.srdb3.conv3.bias", "body.10.srdb3.conv4.weight", "body.10.srdb3.conv4.bias", "body.10.srdb3.conv5.weight", "body.10.srdb3.conv5.bias", "body.11.srdb1.conv1x1.weight", "body.11.srdb1.conv1.weight", "body.11.srdb1.conv1.bias", "body.11.srdb1.conv2.weight", "body.11.srdb1.conv2.bias", "body.11.srdb1.conv3.weight", "body.11.srdb1.conv3.bias", "body.11.srdb1.conv4.weight", "body.11.srdb1.conv4.bias", "body.11.srdb1.conv5.weight", "body.11.srdb1.conv5.bias", "body.11.srdb2.conv1x1.weight", "body.11.srdb2.conv1.weight", "body.11.srdb2.conv1.bias", "body.11.srdb2.conv2.weight", "body.11.srdb2.conv2.bias", "body.11.srdb2.conv3.weight", "body.11.srdb2.conv3.bias", "body.11.srdb2.conv4.weight", "body.11.srdb2.conv4.bias", "body.11.srdb2.conv5.weight", "body.11.srdb2.conv5.bias", "body.11.srdb3.conv1x1.weight", "body.11.srdb3.conv1.weight", "body.11.srdb3.conv1.bias", "body.11.srdb3.conv2.weight", "body.11.srdb3.conv2.bias", "body.11.srdb3.conv3.weight", "body.11.srdb3.conv3.bias", "body.11.srdb3.conv4.weight", "body.11.srdb3.conv4.bias", "body.11.srdb3.conv5.weight", "body.11.srdb3.conv5.bias", "body.12.srdb1.conv1x1.weight", "body.12.srdb1.conv1.weight", "body.12.srdb1.conv1.bias", "body.12.srdb1.conv2.weight", "body.12.srdb1.conv2.bias", "body.12.srdb1.conv3.weight", "body.12.srdb1.conv3.bias", "body.12.srdb1.conv4.weight", "body.12.srdb1.conv4.bias", "body.12.srdb1.conv5.weight", "body.12.srdb1.conv5.bias", "body.12.srdb2.conv1x1.weight", "body.12.srdb2.conv1.weight", "body.12.srdb2.conv1.bias", "body.12.srdb2.conv2.weight", "body.12.srdb2.conv2.bias", "body.12.srdb2.conv3.weight", "body.12.srdb2.conv3.bias", "body.12.srdb2.conv4.weight", "body.12.srdb2.conv4.bias", "body.12.srdb2.conv5.weight", "body.12.srdb2.conv5.bias", "body.12.srdb3.conv1x1.weight", "body.12.srdb3.conv1.weight", "body.12.srdb3.conv1.bias", "body.12.srdb3.conv2.weight", "body.12.srdb3.conv2.bias", "body.12.srdb3.conv3.weight", "body.12.srdb3.conv3.bias", "body.12.srdb3.conv4.weight", "body.12.srdb3.conv4.bias", "body.12.srdb3.conv5.weight", "body.12.srdb3.conv5.bias", "body.13.srdb1.conv1x1.weight", "body.13.srdb1.conv1.weight", "body.13.srdb1.conv1.bias", "body.13.srdb1.conv2.weight", "body.13.srdb1.conv2.bias", "body.13.srdb1.conv3.weight", "body.13.srdb1.conv3.bias", "body.13.srdb1.conv4.weight", "body.13.srdb1.conv4.bias", "body.13.srdb1.conv5.weight", "body.13.srdb1.conv5.bias", "body.13.srdb2.conv1x1.weight", "body.13.srdb2.conv1.weight", "body.13.srdb2.conv1.bias", "body.13.srdb2.conv2.weight", "body.13.srdb2.conv2.bias", "body.13.srdb2.conv3.weight", "body.13.srdb2.conv3.bias", "body.13.srdb2.conv4.weight", "body.13.srdb2.conv4.bias", "body.13.srdb2.conv5.weight", "body.13.srdb2.conv5.bias", "body.13.srdb3.conv1x1.weight", "body.13.srdb3.conv1.weight", "body.13.srdb3.conv1.bias", "body.13.srdb3.conv2.weight", "body.13.srdb3.conv2.bias", "body.13.srdb3.conv3.weight", "body.13.srdb3.conv3.bias", "body.13.srdb3.conv4.weight", "body.13.srdb3.conv4.bias", "body.13.srdb3.conv5.weight", "body.13.srdb3.conv5.bias", "body.14.srdb1.conv1x1.weight", "body.14.srdb1.conv1.weight", "body.14.srdb1.conv1.bias", "body.14.srdb1.conv2.weight", "body.14.srdb1.conv2.bias", "body.14.srdb1.conv3.weight", "body.14.srdb1.conv3.bias", "body.14.srdb1.conv4.weight", "body.14.srdb1.conv4.bias", "body.14.srdb1.conv5.weight", "body.14.srdb1.conv5.bias", "body.14.srdb2.conv1x1.weight", "body.14.srdb2.conv1.weight", "body.14.srdb2.conv1.bias", "body.14.srdb2.conv2.weight", "body.14.srdb2.conv2.bias", "body.14.srdb2.conv3.weight", "body.14.srdb2.conv3.bias", "body.14.srdb2.conv4.weight", "body.14.srdb2.conv4.bias", "body.14.srdb2.conv5.weight", "body.14.srdb2.conv5.bias", "body.14.srdb3.conv1x1.weight", "body.14.srdb3.conv1.weight", "body.14.srdb3.conv1.bias", "body.14.srdb3.conv2.weight", "body.14.srdb3.conv2.bias", "body.14.srdb3.conv3.weight", "body.14.srdb3.conv3.bias", "body.14.srdb3.conv4.weight", "body.14.srdb3.conv4.bias", "body.14.srdb3.conv5.weight", "body.14.srdb3.conv5.bias", "body.15.srdb1.conv1x1.weight", "body.15.srdb1.conv1.weight", "body.15.srdb1.conv1.bias", "body.15.srdb1.conv2.weight", "body.15.srdb1.conv2.bias", "body.15.srdb1.conv3.weight", "body.15.srdb1.conv3.bias", "body.15.srdb1.conv4.weight", "body.15.srdb1.conv4.bias", "body.15.srdb1.conv5.weight", "body.15.srdb1.conv5.bias", "body.15.srdb2.conv1x1.weight", "body.15.srdb2.conv1.weight", "body.15.srdb2.conv1.bias", "body.15.srdb2.conv2.weight", "body.15.srdb2.conv2.bias", "body.15.srdb2.conv3.weight", "body.15.srdb2.conv3.bias", "body.15.srdb2.conv4.weight", "body.15.srdb2.conv4.bias", "body.15.srdb2.conv5.weight", "body.15.srdb2.conv5.bias", "body.15.srdb3.conv1x1.weight", "body.15.srdb3.conv1.weight", "body.15.srdb3.conv1.bias", "body.15.srdb3.conv2.weight", "body.15.srdb3.conv2.bias", "body.15.srdb3.conv3.weight", "body.15.srdb3.conv3.bias", "body.15.srdb3.conv4.weight", "body.15.srdb3.conv4.bias", "body.15.srdb3.conv5.weight", "body.15.srdb3.conv5.bias", "body.16.srdb1.conv1x1.weight", "body.16.srdb1.conv1.weight", "body.16.srdb1.conv1.bias", "body.16.srdb1.conv2.weight", "body.16.srdb1.conv2.bias", "body.16.srdb1.conv3.weight", "body.16.srdb1.conv3.bias", "body.16.srdb1.conv4.weight", "body.16.srdb1.conv4.bias", "body.16.srdb1.conv5.weight", "body.16.srdb1.conv5.bias", "body.16.srdb2.conv1x1.weight", "body.16.srdb2.conv1.weight", "body.16.srdb2.conv1.bias", "body.16.srdb2.conv2.weight", "body.16.srdb2.conv2.bias", "body.16.srdb2.conv3.weight", "body.16.srdb2.conv3.bias", "body.16.srdb2.conv4.weight", "body.16.srdb2.conv4.bias", "body.16.srdb2.conv5.weight", "body.16.srdb2.conv5.bias", "body.16.srdb3.conv1x1.weight", "body.16.srdb3.conv1.weight", "body.16.srdb3.conv1.bias", "body.16.srdb3.conv2.weight", "body.16.srdb3.conv2.bias", "body.16.srdb3.conv3.weight", "body.16.srdb3.conv3.bias", "body.16.srdb3.conv4.weight", "body.16.srdb3.conv4.bias", "body.16.srdb3.conv5.weight", "body.16.srdb3.conv5.bias", "body.17.srdb1.conv1x1.weight", "body.17.srdb1.conv1.weight", "body.17.srdb1.conv1.bias", "body.17.srdb1.conv2.weight", "body.17.srdb1.conv2.bias", "body.17.srdb1.conv3.weight", "body.17.srdb1.conv3.bias", "body.17.srdb1.conv4.weight", "body.17.srdb1.conv4.bias", "body.17.srdb1.conv5.weight", "body.17.srdb1.conv5.bias", "body.17.srdb2.conv1x1.weight", "body.17.srdb2.conv1.weight", "body.17.srdb2.conv1.bias", "body.17.srdb2.conv2.weight", "body.17.srdb2.conv2.bias", "body.17.srdb2.conv3.weight", "body.17.srdb2.conv3.bias", "body.17.srdb2.conv4.weight", "body.17.srdb2.conv4.bias", "body.17.srdb2.conv5.weight", "body.17.srdb2.conv5.bias", "body.17.srdb3.conv1x1.weight", "body.17.srdb3.conv1.weight", "body.17.srdb3.conv1.bias", "body.17.srdb3.conv2.weight", "body.17.srdb3.conv2.bias", "body.17.srdb3.conv3.weight", "body.17.srdb3.conv3.bias", "body.17.srdb3.conv4.weight", "body.17.srdb3.conv4.bias", "body.17.srdb3.conv5.weight", "body.17.srdb3.conv5.bias", "body.18.srdb1.conv1x1.weight", "body.18.srdb1.conv1.weight", "body.18.srdb1.conv1.bias", "body.18.srdb1.conv2.weight", "body.18.srdb1.conv2.bias", "body.18.srdb1.conv3.weight", "body.18.srdb1.conv3.bias", "body.18.srdb1.conv4.weight", "body.18.srdb1.conv4.bias", "body.18.srdb1.conv5.weight", "body.18.srdb1.conv5.bias", "body.18.srdb2.conv1x1.weight", "body.18.srdb2.conv1.weight", "body.18.srdb2.conv1.bias", "body.18.srdb2.conv2.weight", "body.18.srdb2.conv2.bias", "body.18.srdb2.conv3.weight", "body.18.srdb2.conv3.bias", "body.18.srdb2.conv4.weight", "body.18.srdb2.conv4.bias", "body.18.srdb2.conv5.weight", "body.18.srdb2.conv5.bias", "body.18.srdb3.conv1x1.weight", "body.18.srdb3.conv1.weight", "body.18.srdb3.conv1.bias", "body.18.srdb3.conv2.weight", "body.18.srdb3.conv2.bias", "body.18.srdb3.conv3.weight", "body.18.srdb3.conv3.bias", "body.18.srdb3.conv4.weight", "body.18.srdb3.conv4.bias", "body.18.srdb3.conv5.weight", "body.18.srdb3.conv5.bias", "body.19.srdb1.conv1x1.weight", "body.19.srdb1.conv1.weight", "body.19.srdb1.conv1.bias", "body.19.srdb1.conv2.weight", "body.19.srdb1.conv2.bias", "body.19.srdb1.conv3.weight", "body.19.srdb1.conv3.bias", "body.19.srdb1.conv4.weight", "body.19.srdb1.conv4.bias", "body.19.srdb1.conv5.weight", "body.19.srdb1.conv5.bias", "body.19.srdb2.conv1x1.weight", "body.19.srdb2.conv1.weight", "body.19.srdb2.conv1.bias", "body.19.srdb2.conv2.weight", "body.19.srdb2.conv2.bias", "body.19.srdb2.conv3.weight", "body.19.srdb2.conv3.bias", "body.19.srdb2.conv4.weight", "body.19.srdb2.conv4.bias", "body.19.srdb2.conv5.weight", "body.19.srdb2.conv5.bias", "body.19.srdb3.conv1x1.weight", "body.19.srdb3.conv1.weight", "body.19.srdb3.conv1.bias", "body.19.srdb3.conv2.weight", "body.19.srdb3.conv2.bias", "body.19.srdb3.conv3.weight", "body.19.srdb3.conv3.bias", "body.19.srdb3.conv4.weight", "body.19.srdb3.conv4.bias", "body.19.srdb3.conv5.weight", "body.19.srdb3.conv5.bias", "body.20.srdb1.conv1x1.weight", "body.20.srdb1.conv1.weight", "body.20.srdb1.conv1.bias", "body.20.srdb1.conv2.weight", "body.20.srdb1.conv2.bias", "body.20.srdb1.conv3.weight", "body.20.srdb1.conv3.bias", "body.20.srdb1.conv4.weight", "body.20.srdb1.conv4.bias", "body.20.srdb1.conv5.weight", "body.20.srdb1.conv5.bias", "body.20.srdb2.conv1x1.weight", "body.20.srdb2.conv1.weight", "body.20.srdb2.conv1.bias", "body.20.srdb2.conv2.weight", "body.20.srdb2.conv2.bias", "body.20.srdb2.conv3.weight", "body.20.srdb2.conv3.bias", "body.20.srdb2.conv4.weight", "body.20.srdb2.conv4.bias", "body.20.srdb2.conv5.weight", "body.20.srdb2.conv5.bias", "body.20.srdb3.conv1x1.weight", "body.20.srdb3.conv1.weight", "body.20.srdb3.conv1.bias", "body.20.srdb3.conv2.weight", "body.20.srdb3.conv2.bias", "body.20.srdb3.conv3.weight", "body.20.srdb3.conv3.bias", "body.20.srdb3.conv4.weight", "body.20.srdb3.conv4.bias", "body.20.srdb3.conv5.weight", "body.20.srdb3.conv5.bias", "body.21.srdb1.conv1x1.weight", "body.21.srdb1.conv1.weight", "body.21.srdb1.conv1.bias", "body.21.srdb1.conv2.weight", "body.21.srdb1.conv2.bias", "body.21.srdb1.conv3.weight", "body.21.srdb1.conv3.bias", "body.21.srdb1.conv4.weight", "body.21.srdb1.conv4.bias", "body.21.srdb1.conv5.weight", "body.21.srdb1.conv5.bias", "body.21.srdb2.conv1x1.weight", "body.21.srdb2.conv1.weight", "body.21.srdb2.conv1.bias", "body.21.srdb2.conv2.weight", "body.21.srdb2.conv2.bias", "body.21.srdb2.conv3.weight", "body.21.srdb2.conv3.bias", "body.21.srdb2.conv4.weight", "body.21.srdb2.conv4.bias", "body.21.srdb2.conv5.weight", "body.21.srdb2.conv5.bias", "body.21.srdb3.conv1x1.weight", "body.21.srdb3.conv1.weight", "body.21.srdb3.conv1.bias", "body.21.srdb3.conv2.weight", "body.21.srdb3.conv2.bias", "body.21.srdb3.conv3.weight", "body.21.srdb3.conv3.bias", "body.21.srdb3.conv4.weight", "body.21.srdb3.conv4.bias", "body.21.srdb3.conv5.weight", "body.21.srdb3.conv5.bias", "body.22.srdb1.conv1x1.weight", "body.22.srdb1.conv1.weight", "body.22.srdb1.conv1.bias", "body.22.srdb1.conv2.weight", "body.22.srdb1.conv2.bias", "body.22.srdb1.conv3.weight", "body.22.srdb1.conv3.bias", "body.22.srdb1.conv4.weight", "body.22.srdb1.conv4.bias", "body.22.srdb1.conv5.weight", "body.22.srdb1.conv5.bias", "body.22.srdb2.conv1x1.weight", "body.22.srdb2.conv1.weight", "body.22.srdb2.conv1.bias", "body.22.srdb2.conv2.weight", "body.22.srdb2.conv2.bias", "body.22.srdb2.conv3.weight", "body.22.srdb2.conv3.bias", "body.22.srdb2.conv4.weight", "body.22.srdb2.conv4.bias", "body.22.srdb2.conv5.weight", "body.22.srdb2.conv5.bias", "body.22.srdb3.conv1x1.weight", "body.22.srdb3.conv1.weight", "body.22.srdb3.conv1.bias", "body.22.srdb3.conv2.weight", "body.22.srdb3.conv2.bias", "body.22.srdb3.conv3.weight", "body.22.srdb3.conv3.bias", "body.22.srdb3.conv4.weight", "body.22.srdb3.conv4.bias", "body.22.srdb3.conv5.weight", "body.22.srdb3.conv5.bias", "conv_body.weight", "conv_body.bias", "conv_up1.weight", "conv_up1.bias", "conv_up2.weight", "conv_up2.bias", "conv_hr.weight", "conv_hr.bias", "conv_last.weight", "conv_last.bias". Unexpected key(s) in state_dict: "layer0.conv0.weight", "layer0.conv0.bias", "layer0.conv1.weight_orig", "layer0.conv1.weight_u", "layer0.conv1.weight_v", "layer0.conv2.weight_orig", "layer0.conv2.weight_u", "layer0.conv2.weight_v", "layer0.conv3.weight_orig", "layer0.conv3.weight_u", "layer0.conv3.weight_v", "layer0.gating.weight_orig", "layer0.gating.weight_u", "layer0.gating.weight_v", "layer0.attn_1.W.0.weight", "layer0.attn_1.W.0.bias", "layer0.attn_1.W.1.weight", "layer0.attn_1.W.1.bias", "layer0.attn_1.W.1.running_mean", "layer0.attn_1.W.1.running_var", "layer0.attn_1.W.1.num_batches_tracked", "layer0.attn_1.theta.weight", "layer0.attn_1.phi.weight", "layer0.attn_1.phi.bias", "layer0.attn_1.psi.weight", "layer0.attn_1.psi.bias", "layer0.attn_2.W.0.weight", "layer0.attn_2.W.0.bias", "layer0.attn_2.W.1.weight", "layer0.attn_2.W.1.bias", "layer0.attn_2.W.1.running_mean", "layer0.attn_2.W.1.running_var", "layer0.attn_2.W.1.num_batches_tracked", "layer0.attn_2.theta.weight", "layer0.attn_2.phi.weight", "layer0.attn_2.phi.bias", "layer0.attn_2.psi.weight", "layer0.attn_2.psi.bias", "layer0.attn_3.W.0.weight", "layer0.attn_3.W.0.bias", "layer0.attn_3.W.1.weight", "layer0.attn_3.W.1.bias", "layer0.attn_3.W.1.running_mean", "layer0.attn_3.W.1.running_var", "layer0.attn_3.W.1.num_batches_tracked", "layer0.attn_3.theta.weight", "layer0.attn_3.phi.weight", "layer0.attn_3.phi.bias", "layer0.attn_3.psi.weight", "layer0.attn_3.psi.bias", "layer0.cat_1.convU.weight_orig", "layer0.cat_1.convU.weight_u", "layer0.cat_1.convU.weight_v", "layer0.cat_2.convU.weight_orig", "layer0.cat_2.convU.weight_u", "layer0.cat_2.convU.weight_v", "layer0.cat_3.convU.weight_orig", "layer0.cat_3.convU.weight_u", "layer0.cat_3.convU.weight_v", "layer0.conv4.weight_orig", "layer0.conv4.weight_u", "layer0.conv4.weight_v", "layer0.conv5.weight_orig", "layer0.conv5.weight_u", "layer0.conv5.weight_v", "layer0.conv6.weight_orig", "layer0.conv6.weight_u", "layer0.conv6.weight_v", "layer0.conv7.weight_orig", "layer0.conv7.weight_u", "layer0.conv7.weight_v", "layer0.conv8.weight_orig", "layer0.conv8.weight_u", "layer0.conv8.weight_v", "layer0.conv9.weight", "layer0.conv9.bias", "layer1.conv0.weight", "layer1.conv0.bias", "layer1.conv1.weight_orig", "layer1.conv1.weight_u", "layer1.conv1.weight_v", "layer1.conv2.weight_orig", "layer1.conv2.weight_u", "layer1.conv2.weight_v", "layer1.conv3.weight_orig", "layer1.conv3.weight_u", "layer1.conv3.weight_v", "layer1.gating.weight_orig", "layer1.gating.weight_u", "layer1.gating.weight_v", "layer1.attn_1.W.0.weight", "layer1.attn_1.W.0.bias", "layer1.attn_1.W.1.weight", "layer1.attn_1.W.1.bias", "layer1.attn_1.W.1.running_mean", "layer1.attn_1.W.1.running_var", "layer1.attn_1.W.1.num_batches_tracked", "layer1.attn_1.theta.weight", "layer1.attn_1.phi.weight", "layer1.attn_1.phi.bias", "layer1.attn_1.psi.weight", "layer1.attn_1.psi.bias", "layer1.attn_2.W.0.weight", "layer1.attn_2.W.0.bias", "layer1.attn_2.W.1.weight", "layer1.attn_2.W.1.bias", "layer1.attn_2.W.1.running_mean", "layer1.attn_2.W.1.running_var", "layer1.attn_2.W.1.num_batches_tracked", "layer1.attn_2.theta.weight", "layer1.attn_2.phi.weight", "layer1.attn_2.phi.bias", "layer1.attn_2.psi.weight", "layer1.attn_2.psi.bias", "layer1.attn_3.W.0.weight", "layer1.attn_3.W.0.bias", "layer1.attn_3.W.1.weight", "layer1.attn_3.W.1.bias", "layer1.attn_3.W.1.running_mean", "layer1.attn_3.W.1.running_var", "layer1.attn_3.W.1.num_batches_tracked", "layer1.attn_3.theta.weight", "layer1.attn_3.phi.weight", "layer1.attn_3.phi.bias", "layer1.attn_3.psi.weight", "layer1.attn_3.psi.bias", "layer1.cat_1.convU.weight_orig", "layer1.cat_1.convU.weight_u", "layer1.cat_1.convU.weight_v", "layer1.cat_2.convU.weight_orig", "layer1.cat_2.convU.weight_u", "layer1.cat_2.convU.weight_v", "layer1.cat_3.convU.weight_orig", "layer1.cat_3.convU.weight_u", "layer1.cat_3.convU.weight_v", "layer1.conv4.weight_orig", "layer1.conv4.weight_u", "layer1.conv4.weight_v", "layer1.conv5.weight_orig", "layer1.conv5.weight_u", "layer1.conv5.weight_v", "layer1.conv6.weight_orig", "layer1.conv6.weight_u", "layer1.conv6.weight_v", "layer1.conv7.weight_orig", "layer1.conv7.weight_u", "layer1.conv7.weight_v", "layer1.conv8.weight_orig", "layer1.conv8.weight_u", "layer1.conv8.weight_v", "layer1.conv9.weight", "layer1.conv9.bias".

Any suggestions on how to resume the gan training from last saved state?
mabubakarsaleem commented 4 months ago

Have tou find the solution,As i got the issues regarding the resuming of the model. How can we resume the model?