model = ContextEncoder() print(model) pretrained_dict = torch.load('E:/fin/models/gen.pth') print(pretrained_dict.keys())上述语句输出为:ContextEncoder( (encoder): Sequential( (0): Conv2d(4, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (1): LeakyReLU(negative_slope=0.2, inplace=True) (2): Conv2d(64, 128, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (3): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (4): LeakyReLU(negative_slope=0.2, inplace=True) (5): Conv2d(128, 256, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (6): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (7): LeakyReLU(negative_slope=0.2, inplace=True) (8): Conv2d(256, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (9): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (10): LeakyReLU(negative_slope=0.2, inplace=True) (11): Conv2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (12): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (13): LeakyReLU(negative_slope=0.2, inplace=True) (14): Conv2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (15): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (16): LeakyReLU(negative_slope=0.2, inplace=True) (17): Conv2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (18): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (19): LeakyReLU(negative_slope=0.2, inplace=True) (20): Conv2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (21): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (22): LeakyReLU(negative_slope=0.2, inplace=True) (23): Conv2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (24): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (25): LeakyReLU(negative_slope=0.2, inplace=True) ) (decoder): Sequential( (0): ConvTranspose2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (1): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (2): ReLU(inplace=True) (3): ConvTranspose2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (4): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (5): ReLU(inplace=True) (6): ConvTranspose2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (7): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (8): ReLU(inplace=True) (9): ConvTranspose2d(512, 512, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (10): BatchNorm2d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (11): ReLU(inplace=True) (12): ConvTranspose2d(512, 256, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (13): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (14): ReLU(inplace=True) (15): ConvTranspose2d(256, 128, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (16): BatchNorm2d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (17): ReLU(inplace=True) (18): ConvTranspose2d(128, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (19): BatchNorm2d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (20): ReLU(inplace=True) (21): ConvTranspose2d(64, 3, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1)) (22): Sigmoid() ) ) dict_keys(['iteration', 'generator'])
最新发布