lithostark

splitnet

Code for SplitNet paper

DSD-training

Implementation of DSD: Dense-Sparse-Dense Training for Deep Neural Networks in Pytorch.

autograd

Efficiently computes derivatives of numpy code.

Gradient-Projection-Method

final exam matlab file for gradient projection method

BregmanLearning

Optimizing neural networks via an inverse scale space flow.

codedapertures

a python package for generating coded aperture patterns

stylegan3

Official PyTorch implementation of StyleGAN3

SAR_specific_models

GraN-DAG

ICML2021-Gem

Official code for the ICML 2021 paper "Generative Causal Explanations for Graph Neural Networks."

GNNPapers

Must-read papers on graph neural networks (GNN)

stacked-capsule-networks

Pytorch Implementation of the Stacked Capsule Autoencoders

graphtransformer

Graph Transformer Architecture. Source code for "A Generalization of Transformer Networks to Graphs", DLG-AAAI'21.

MultiDigitMNIST

Combine multiple MNIST digits to create datasets with 100/1000 classes for few-shot learning/meta-learning

psgd_torch

Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner and more)

MCMIL

A multiclass multiple instance learning method. Only need to know whether any instance of a class exists or not in a sample. Demonstration on multiple object detection and localization.

Dassl.pytorch

A PyTorch toolbox for domain generalization, domain adaptation and semi-supervised learning.

MultiObjectiveOptimization

Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"

xcit

Official code Cross-Covariance Image Transformer (XCiT)

pyprobml

Python code for "Probabilistic Machine learning" book by Kevin Murphy

transferlearning

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

congruency

[TPAMI 2019] The implementation for "Direction Concentration Learning: Enhancing Congruency in Machine Learning"

WMMSE-deep-unfolding

pytorch_geometric

Graph Neural Network Library for PyTorch

DAGCN

Dual Attention Graph Convolutional Network

pytorch-faster-rcnn

pytorch1.0 updated. Support cpu test and demo. (Use detectron2, it's a masterpiece)

SpinNet

[CVPR 2021] SpinNet: Learning a General Surface Descriptor for 3D Point Cloud Registration

PointFlow

PointFlow : 3D Point Cloud Generation with Continuous Normalizing Flows

3D-point-capsule-networks

3D Point Capsule Networks

qecnetworks

Official implementation of the ECCV 2020 Oral Paper "Quaternion Equivariant Capsule Networks for 3D Point Clouds"