Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
thon
outputs.reconstruction.data.shape
torch.Size([1, 3, 880, 1072])
We need to squeeze the output and get rid of axis 0, clip the values, then convert it to be numpy float.