A GPT-2 ChatBot implemented using Pytorch and Huggingface-transformers
pip install -r requirements.txt
python load_data.py
you can train the model from a initial state
python main.py --mode="train"
also you can train a model from a specific checkpoint (eg. best.ckpt)
python main.py --mode="train" --ckpt_name="best"
python main.py --mode="infer" --ckpt_name="best"
This repo provides code for QB-Norm (Cross Modal Retrieval with Querybank Normalisation) Usage example python dynamic_inverted_softmax.py --sims_train
Bayesian-Torch is a library of neural network layers and utilities extending the core of PyTorch to enable the user to perform stochastic variational inference in Bayesian deep neural networks. Bayes
CyberBattleSim April 8th, 2021: See the announcement on the Microsoft Security Blog. CyberBattleSim is an experimentation research platform to investi
Sharpened Cosine Similarity A layer implementation for PyTorch Install At your c
Summary Explorer Summary Explorer is a tool to visually inspect the summaries from several state-of-the-art neural summarization models across multipl
FInite volume Neural Network (FINN) This repository contains the PyTorch code for models, training, and testing, and Python code for data generation t
Keras RetinaNet Keras implementation of RetinaNet object detection as described in Focal Loss for Dense Object Detection by Tsung-Yi Lin, Priya Goyal,
Length-Controllable Image Captioning (ECCV2020) This repo provides the implemetation of the paper Length-Controllable Image Captioning. Install conda
Modeling distributions on Riemannian manifolds is a crucial component in understanding non-Euclidean data that arises, e.g., in physics and geology. The budding approaches in this space are limited b
MADT: Offline Pre-trained Multi-Agent Decision Transformer A link to our paper can be found on Arxiv. Overview Official codebase for Offline Pre-train
GNDC For submission to IEEE TKDE. Overview Here we provide the implementation of GND-Nets (Graph Neural Diffusion Networks) in TensorFlow. The reposit
LOCKER This is the pytorch implementation of this paper: Locally Constrained Self-Attentive Sequential Recommendation. Zhankui He, Handong Zhao, Zhe L
Applicator Kit for Modo Applicator Kit for Modo allow you to apply Apple ARKit Face Tracking data from your iPhone or iPad with a TrueDepth camera to
Variational Graph Auto-encoder in Pytorch This repository implements variational graph auto-encoder by Thomas Kipf. For details of the model, refer to
ASSL This repository is for a new network pruning method (Aligned Structured Sparsity Learning, ASSL) for efficient single image super-resolution (SR)
title emoji colorFrom colorTo sdk app_file pinned Transport_Mode_Detector 🚀 purple yellow gradio app.py false Configuration title: string Display tit
Implémentation en pyhton de l'article Depixelizing pixel art de Johannes Kopf et Dani Lischinski
QuakeLabeler Quake Labeler was born from the need for seismologists and developers who are not AI specialists to easily, quickly, and independently bu
Generative Handwriting Demo using TensorFlow An attempt to implement the random handwriting generation portion of Alex Graves' paper. See my blog post
Prototypical Networks for Few shot Learning in PyTorch Simple alternative Implementation of Prototypical Networks for Few Shot Learning (paper, code)
36 Dec 29, 2022
210 Jan 04, 2023
1.5k Dec 25, 2022
203 Nov 30, 2022
42 Aug 14, 2022
20 Dec 18, 2022
4.3k Jan 01, 2023
53 Nov 13, 2022
61 Nov 28, 2022
51 Dec 21, 2022
3 Aug 08, 2022
8 Jul 30, 2022
3 Aug 24, 2021
215 Jan 02, 2023
47 Nov 28, 2022
3 Jan 16, 2022
3 May 29, 2022
15 Nov 04, 2022
686 Nov 24, 2022
835 Jan 08, 2023