PyTorch-Image-Retrieval icon indicating copy to clipboard operation
PyTorch-Image-Retrieval copied to clipboard

verified publisher icon leeesangwon
→

Metadata

A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).

Results 4 PyTorch-Image-Retrieval issues
Sort by recently updated
recently updated
newest added

The following problems occur when running: Process finished with exit code -1073741819 (0xC0000005)

Process finished with exit code -1073741819 (0xC0000005)

xiaoweidao avatar xiaoweidao

RuntimeError: CUDA out of memory. Tried to allocate 240.00 MiB (GPU 0; 14.76 GiB total capacity; 13.54 GiB already allocated; 73.75 MiB free; 13.63 GiB reserved in total by PyTorch)

I am new to using `PyTorch`. I am using `Google Colaboratory`, `Accelerator` is `GPU`. Attempted Solutions (same error): - `torch.cuda.empty_cache()`, [suggested here][1]. - `torch.cuda.memory_summary(device=None, abbreviated=False)`, [suggested here][1]. - `batch_size=1` for...

danielbellhv avatar danielbellhv

what about the mAP?

onlinehuazai avatar onlinehuazai

Bump torch from 1.0.0 to 2.2.0

Bumps [torch](https://github.com/pytorch/pytorch) from 1.0.0 to 2.2.0. Release notes Sourced from torch's releases. PyTorch 2.2: FlashAttention-v2, AOTInductor PyTorch 2.2 Release Notes Highlights Backwards Incompatible Changes Deprecations New Features Improvements Bug fixes...

dependabot[bot] avatar dependabot[bot]

dependencies

← Metadata

273
Stars
55
Forks
Watchers

Owner

verified publisher icon leeesangwon

Metadata

A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).

Back