Skip to content
#

pytorch

Here are 19,160 public repositories matching this topic...

transformers
fcakyon
fcakyon commented Dec 6, 2021

🚀 Feature request

Make CLIPFeatureExtractor (or any FeatureExtractor in general) accept batch of images as torch.Tensor.

Motivation

Currently batch of images as torch.Tensor are not treated as a batch, it has to be a List[torch.Tensor] but it is not the case when using native Pytorch DataLoader. Can we update this line so that it accepts batches as torch.Tensor. Maybe we ca

pytorch-lightning
awaelchli
awaelchli commented Dec 8, 2021

Proposed refactor

Deprecate tpu_global_core_rank in favor of global_rank.

Pitch

It "looks" like tpu_global_core_rank in the TPUPlugin is just another name for what is known as the global rank. After #10896 we should investigate whether the two are equivalent and revert back to using the unified global_rank property exclusively.


chan4cc
chan4cc commented Apr 26, 2021

New Operator

Describe the operator

Why is this operator necessary? What does it accomplish?

This is a frequently used operator in tensorflow/keras

Can this operator be constructed using existing onnx operators?

If so, why not add it as a function?

I don't know.

Is this operator used by any model currently? Which one?

Are you willing to contribute it?

datasets
danieldeutsch
danieldeutsch commented Jun 2, 2021

Is your feature request related to a problem? Please describe.
I typically used compressed datasets (e.g. gzipped) to save disk space. This works fine with AllenNLP during training because I can write my dataset reader to load the compressed data. However, the predict command opens the file and reads lines for the Predictor. This fails when it tries to load data from my compressed files.

nni

Improve this page

Add a description, image, and links to the pytorch topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the pytorch topic, visit your repo's landing page and select "manage topics."

Learn more