Closed qkqhd222 closed 1 year ago
Thanks for releasing such a great package. When I import dkm at my custom script, just importing makes additional memory occupancy on gpu 0.
Moving the following line to tensor_to_pil() seems to fix this.
imagenet_mean = torch.tensor([0.485, 0.456, 0.406]).to(device) imagenet_std = torch.tensor([0.229, 0.224, 0.225]).to(device)
(https://github.com/Parskatt/DKM/blob/main/dkm/utils/utils.py#L294)
Could you check?
Hi, thanks for noticing! Could you submit a pull request with it fixed?
Thanks for releasing such a great package. When I import dkm at my custom script, just importing makes additional memory occupancy on gpu 0.
Moving the following line to tensor_to_pil() seems to fix this.
imagenet_mean = torch.tensor([0.485, 0.456, 0.406]).to(device) imagenet_std = torch.tensor([0.229, 0.224, 0.225]).to(device)
(https://github.com/Parskatt/DKM/blob/main/dkm/utils/utils.py#L294)
Could you check?