styler00dollar

🐛 [Bug] conv2d of sum results in "Expected orig_dims.nbDims > 2 to be true but got false. Unable to create convolution layer from node"

4

## To Reproduce ```python # https://github.com/bilibili/ailab/blob/main/Real-CUGAN/VapourSynth/upcunet_v3_vs.py import torch import torch_tensorrt from torch import nn as nn from torch.nn import functional as F import os,sys import numpy as np from tqdm...

Multithreading / Threaded inference does not work with Python API

1

## Description I want to have multithreaded inference, since normal inference results in about 10% GPU usage, but it does not seem to work. ## Steps To Reproduce Here is...

PyTorch Lightning support

1

Is this project supposed to work with PyTorch Lightning? I tried to add it and it seems like in terms of speed nothing changed after adding `@accelerate_model()` and `@accelerate_dataset()`. The...

Running demo code results in "LinAlgError: SVD did not converge" or "ValueError: array must not contain infs or NaNs"

1

Like I already mentioned in [Issue 13](https://github.com/musco-ai/musco-pytorch/issues/13#issuecomment-812170929), the demo code seems to crash with an error. ``` from torchvision.models import resnet50 from flopco import FlopCo from musco.pytorch import CompressorVBMF, CompressorPR,...

Feature request: Add ensembling

4

Ensembling is used in frame interpolation to drastically improve visual quality by using different predictions and generating a mean. [Here](https://arxiv.org/pdf/2011.01280.pdf) is a paper talking about ensembling.  Rife does use...

🐛 [Bug] nn.PixelShuffle crashes torch_tensorrt.compile

2

## Bug Description `nn.PixelShuffle` should work, but it does not. ## To Reproduce ```python from torch import nn as nn import torch class mynet(nn.Module): def __init__(self): super(mynet, self).__init__() self.conv =...

🐛 [Bug] nn.PReLU crashes torch_tensorrt.compile

5

## Bug Description Compiling model fails if `nn.PReLU` is used. It works fine with `nn.ReLU`. ## To Reproduce Just run this code ```python from torch import nn as nn from...

I like this repo, but I noticed that it does not work with a recent version of kornia. `RandomMixUp` and `RandomCutMix` got replaced with [RandomMixUpV2](https://kornia.readthedocs.io/en/latest/augmentation.module.html#kornia.augmentation.RandomMixUpV2) and [RandomCutMixV2](https://kornia.readthedocs.io/en/latest/augmentation.module.html#kornia.augmentation.RandomCutMixV2) in kornia. Would...