A Pytorch Implementation for Compact Bilinear Pooling.

Last update: Dec 23, 2022

Overview

CompactBilinearPooling-Pytorch

A Pytorch Implementation for Compact Bilinear Pooling. Adapted from tensorflow_compact_bilinear_pooling

Prerequisites

pip install pytorch_fft

Usage

from torch import nn
from torch.autograd import Variable
from CompactBilinearPooling import CompactBilinearPooling

bottom1 = Variable(torch.randn(128, 512, 14, 14)).cuda()
bottom2 = Variable(torch.randn(128, 512, 14, 14)).cuda()

layer = CompactBilinearPooling(512, 512, 8000)
layer.cuda()
layer.train()

out = layer(bottom1, bottom2)

Reference

Yang Gao, et al. "Compact Bilinear Pooling." in Proceedings of IEEE Conference on Computer Vision and Pattern Recognition (2016).
Akira Fukui, et al. "Multimodal Compact Bilinear Pooling for Visual Question Answering and Visual Grounding." arXiv preprint arXiv:1606.01847 (2016).

Owner

GitHub Repository

High-level batteries-included neural network training library for Pytorch

Pywick High-Level Training framework for Pytorch Pywick is a high-level Pytorch training framework that aims to get you up and running quickly with st

382 Dec 06, 2022

higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps.

higher is a library providing support for higher-order optimization, e.g. through unrolled first-order optimization loops, of "meta" aspects of these

1.5k Jan 03, 2023

Fast, general, and tested differentiable structured prediction in PyTorch

Torch-Struct: Structured Prediction Library A library of tested, GPU implementations of core structured prediction algorithms for deep learning applic

1.1k Jan 07, 2023

Code for paper "Energy-Constrained Compression for Deep Neural Networks via Weighted Sparse Projection and Layer Input Masking"

model_based_energy_constrained_compression Code for paper "Energy-Constrained Compression for Deep Neural Networks via Weighted Sparse Projection and

16 Jun 15, 2022

A Pytorch Implementation for Compact Bilinear Pooling.

Related tags

Overview

CompactBilinearPooling-Pytorch

Prerequisites

Usage

Reference

Owner

High-level batteries-included neural network training library for Pytorch

higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps.

Fast, general, and tested differentiable structured prediction in PyTorch

Code for paper "Energy-Constrained Compression for Deep Neural Networks via Weighted Sparse Projection and Layer Input Masking"

PyTorch toolkit for biomedical imaging

Official implementations of EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis.

Unofficial PyTorch implementation of DeepMind's Perceiver IO with PyTorch Lightning scripts for distributed training

Training PyTorch models with differential privacy

OptNet: Differentiable Optimization as a Layer in Neural Networks

ONNX Runtime for PyTorch accelerates PyTorch model training using ONNX Runtime.

S3-plugin is a high performance PyTorch dataset library to efficiently access datasets stored in S3 buckets.

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

Implements pytorch code for the Accelerated SGD algorithm.

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch

This is an differentiable pytorch implementation of SIFT patch descriptor.

PyTorch extensions for fast R&D prototyping and Kaggle farming

A tiny package to compare two neural networks in PyTorch

A tutorial on "Bayesian Compression for Deep Learning" published at NIPS (2017).