Structural Similarity Index (SSIM) Universal Quality Image Index (UQI) Multi-scale Structural Similarity Index (MS-SSIM) Erreur Relative Globale Adimensionnelle de Synthèse (ERGAS) Spatial Correlation Coefficient (SCC) Relative Average Spectral Error (RASE) Spectral Angle Mapper (SAM) Spectral Distorti...
torchmetrics/image/ms_ssim.py Outdated Show resolved torchmetrics/image/ms_ssim.py Show resolved torchmetrics/functional/image/ms_ssim.py Show resolved torchmetrics/functional/image/ms_ssim.py Outdated Show resolved Borda and others added 3 commits December 15, 2021 18:50 fix badge 74202eb...
Tensor: """ This is the equivalent of torch.repeat_interleave(x, dim=1, repeats=n_rep). The hidden states go from (batch, num_key_value_heads, seqlen, head_dim) to (batch, num_attention_heads, seqlen, head_dim) """ batch, num_key_value_heads, slen, head_dim = hidden_states...