Torchsort Pure PyTorch implementation of Fast Differentiable Sorting and Ranking (Blondel et al.). Much of the code is copied from the original Numpy implementation at google-research/fast-soft-sort, with the isotonic regression solver rewritten as a PyTorch C++ and CUDA extension. Install pip install torchsort To build the CUDA extension you will need the CUDA toolchain installed. If youwant to build in an environment without a CUDA runtime (e.g. docker), you willneed to export the environment variableTORCH_CUDA_ARCH_LIST=”Pascal;Volta;Turing” before installing. Usage torchsort exposes […]

focal-frequency-loss This repository will provide the official code for the following paper: Focal Frequency Loss for Generative ModelsLiming Jiang, Bo Dai, Wayne Wu and Chen Change LoyarXiv preprint, 2020. Abstract: Despite the remarkable success of generative models in creating photorealistic images using deep neural networks, gaps could still exist between the real and generated images, especially in the frequency domain. In this study, we find that narrowing the frequency domain gap can ameliorate the image synthesis quality further. To this […]

ParallelStencil.jl ParallelStencil empowers domain scientists to write architecture-agnostic high-level code for parallel high-performance stencil computations on GPUs and CPUs. Performance similar to CUDA C can be achieved, which is typically a large improvement over the performance reached when using only CUDA.jl Array programming. For example, a 2-D shallow ice solver presented at JuliaCon 2020 [1] achieved a nearly 20 times better performance than a corresponding CUDA.jl Array programming implementation; in absolute terms, it reached 70% of the theoretical upper performance […]

Detectron2 This repo contains the training configurations, code and trained models trained on PubLayNet dataset using Detectron2 implementation.PubLayNet is a very large dataset for document layout analysis (document segmentation). It can be used to trained semantic segmentation/Object detection models. NOTE Models are trained on a portion of the dataset (train-0.zip, train-1.zip, train-2.zip, train-3.zip) Trained on total 191,832 images Models are evaluated on dev.zip (~11,000 images) Backbone pretrained on COCO dataset is used but trained from scratch on PubLayNet dataset Trained […]