Skip to content

Commit c194934

Browse files
author
pytorchbot
committed
2024-11-28 nightly release (19fef3d)
1 parent 83e4bcc commit c194934

File tree

6 files changed

+9
-6
lines changed

6 files changed

+9
-6
lines changed

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ docs/source/auto_examples/
1616
docs/source/gen_modules/
1717
docs/source/generated/
1818
docs/source/models/generated/
19+
docs/source/sg_execution_times.rst
1920
# pytorch-sphinx-theme gets installed here
2021
docs/src
2122

@@ -42,3 +43,5 @@ xcuserdata/
4243
# direnv
4344
.direnv
4445
.envrc
46+
47+
scripts/release_notes/data.json

references/depth/stereo/utils/losses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def make_gaussian_kernel(kernel_size: int, sigma: float) -> torch.Tensor:
1313
y = torch.arange(kernel_size, dtype=torch.float32)
1414
x = x - (kernel_size - 1) / 2
1515
y = y - (kernel_size - 1) / 2
16-
x, y = torch.meshgrid(x, y)
16+
x, y = torch.meshgrid(x, y, indexing="ij")
1717
grid = (x**2 + y**2) / (2 * sigma**2)
1818
kernel = torch.exp(-grid)
1919
kernel = kernel / kernel.sum()

references/video_classification/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def main(args):
164164

165165
if args.cache_dataset and os.path.exists(cache_path):
166166
print(f"Loading dataset_train from {cache_path}")
167-
dataset, _ = torch.load(cache_path, weights_only=True)
167+
dataset, _ = torch.load(cache_path, weights_only=False)
168168
dataset.transform = transform_train
169169
else:
170170
if args.distributed:
@@ -201,7 +201,7 @@ def main(args):
201201

202202
if args.cache_dataset and os.path.exists(cache_path):
203203
print(f"Loading dataset_test from {cache_path}")
204-
dataset_test, _ = torch.load(cache_path, weights_only=True)
204+
dataset_test, _ = torch.load(cache_path, weights_only=False)
205205
dataset_test.transform = transform_test
206206
else:
207207
if args.distributed:

torchvision/datasets/lsun.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def __len__(self) -> int:
5555

5656

5757
class LSUN(VisionDataset):
58-
"""`LSUN <https://www.yf.io/p/lsun>`_ dataset.
58+
"""`LSUN <https://paperswithcode.com/dataset/lsun>`_ dataset.
5959
6060
You will need to install the ``lmdb`` package to use this dataset: run
6161
``pip install lmdb``

torchvision/models/googlenet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def eager_outputs(self, x: Tensor, aux2: Tensor, aux1: Optional[Tensor]) -> Goog
171171

172172
def forward(self, x: Tensor) -> GoogLeNetOutputs:
173173
x = self._transform_input(x)
174-
x, aux1, aux2 = self._forward(x)
174+
x, aux2, aux1 = self._forward(x)
175175
aux_defined = self.training and self.aux_logits
176176
if torch.jit.is_scripting():
177177
if not aux_defined:

torchvision/models/maxvit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def _make_block_input_shapes(input_size: Tuple[int, int], n_blocks: int) -> List
4040

4141

4242
def _get_relative_position_index(height: int, width: int) -> torch.Tensor:
43-
coords = torch.stack(torch.meshgrid([torch.arange(height), torch.arange(width)]))
43+
coords = torch.stack(torch.meshgrid([torch.arange(height), torch.arange(width)], indexing="ij"))
4444
coords_flat = torch.flatten(coords, 1)
4545
relative_coords = coords_flat[:, :, None] - coords_flat[:, None, :]
4646
relative_coords = relative_coords.permute(1, 2, 0).contiguous()

0 commit comments

Comments
 (0)