Oh I see,  here’s the result, thanks again
Info for Conv2d(8, 8, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2)).
stride: (96768, 12096, 63, 1)
input size:  torch.Size([64, 8, 192, 63])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for Conv2d(8, 16, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2)).
stride: (24576, 3072, 32, 1)
input size:  torch.Size([64, 8, 96, 32])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for Conv2d(16, 16, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2)).
stride: (12288, 768, 16, 1)
input size:  torch.Size([64, 16, 48, 16])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for Conv2d(16, 32, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2)).
stride: (3072, 192, 8, 1)
input size:  torch.Size([64, 16, 24, 8])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for Conv2d(32, 64, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2)).
stride: (1536, 48, 4, 1)
input size:  torch.Size([64, 32, 12, 4])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for ConvTranspose2d(64, 32, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2), output_padding=(1, 1)).
stride: (768, 12, 2, 1)
input size:  torch.Size([64, 64, 6, 2])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for ConvTranspose2d(64, 16, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2), output_padding=(1, 1)).
stride: (3072, 48, 4, 1)
input size:  torch.Size([64, 64, 12, 4])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for ConvTranspose2d(32, 16, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2), output_padding=(1, 1)).
stride: (6144, 192, 8, 1)
input size:  torch.Size([64, 32, 24, 8])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for ConvTranspose2d(32, 8, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2), output_padding=(1, 1)).
stride: (24576, 768, 16, 1)
input size:  torch.Size([64, 32, 48, 16])
input offset:  0
input device:  cuda:7
input layout: torch.strided
Info for ConvTranspose2d(16, 8, kernel_size=(5, 5), stride=(2, 2), padding=(2, 2), output_padding=(1, 1)).
stride: (49152, 3072, 32, 1)
input size:  torch.Size([64, 16, 96, 32])
input offset:  0
input device:  cuda:7
input layout: torch.strided