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"
One-shot Affordance Detection PyTorch implementation of our one-shot affordance detection models. This repository contains PyTorch evaluation code, tr
MUST-GAN Code | paper The Pytorch implementation of our CVPR2021 paper "MUST-GAN: Multi-level Statistics Transfer for Self-driven Person Image Generat
PPO Pytorch C++ This is an implementation of the proximal policy optimization algorithm for the C++ API of Pytorch. It uses a simple TestEnvironment t
Semi-supervised Implicit Scene Completion from Sparse LiDAR Paper Created by Pengfei Li, Yongliang Shi, Tianyu Liu, Hao Zhao, Guyue Zhou and YA-QIN ZH
Detecting Twenty-thousand Classes using Image-level Supervision Detic: A Detector with image classes that can use image-level labels to easily train d
VIN: Value Iteration Networks This is an implementation of Value Iteration Networks (VIN) in PyTorch to reproduce the results.(TensorFlow version) Key
TCMR: Beyond Static Features for Temporally Consistent 3D Human Pose and Shape from a Video Qualtitative result Paper teaser video Introduction This r
DIGIT-INTERFACE Python interface for the DIGIT tactile sensor. For updates and discussions please join the #DIGIT channel at the www.touch-sensing.org
BBQ Repository for the Bias Benchmark for QA dataset. Authors: Alicia Parrish, Angelica Chen, Nikita Nangia, Vishakh Padmakumar, Jason Phang, Jana Tho
Documentation pythae This library implements some of the most common (Variational) Autoencoder models. In particular it provides the possibility to pe
TL;DR: CrossVIS (Crossover Learning for Fast Online Video Instance Segmentation) proposes a novel crossover learning paradigm to fully leverage rich c
M3DSSD: Monocular 3D Single Stage Object Detector Setup pytorch 0.4.1 Preparation Download the full KITTI detection dataset. Then place a softlink (or
This is an unofficial PyTorch implementation of Meta Pseudo Labels. The official Tensorflow implementation is here.
This version of the newest PyTorch-based StyleGAN2-ada is intended mostly for fellow artists, who rarely look at scientific metrics, but rather need a working creative tool. Tested on Python 3.7 + Py
Uniformer - Pytorch Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification ta
Self-Supervised Reward Regression (SSRR) Codebase for CoRL 2021 paper "Learning from Suboptimal Demonstration via Self-Supervised Reward Regression "
vs-RealEsrganAnime-tensorrt-docker Using image super resolution models with vapoursynth and speeding them up with TensorRT. Also a docker image since
Build an Amazon SageMaker Pipeline to Transform Raw Texts to A Knowledge Graph This repository provides a pipeline to create a knowledge graph from ra
JAX Gravity This repo contains a demo of how to use JAX to create a simple gravity simulation. It uses JAX's experimental ode package to solve the dif
tight-binder Introduction General purpose tight-binding code for electronic structure calculations based on the Slater-Koster approximation. The code
46 Dec 12, 2022
46 Dec 26, 2022
59 Dec 09, 2022
114 Nov 30, 2022
1.3k Jan 04, 2023
215 Dec 07, 2022
215 Jan 06, 2023
18 Nov 18, 2022
1.1k Jan 02, 2023
79 Nov 25, 2022
64 Dec 27, 2022
320 Jan 08, 2023
170 Nov 16, 2022
90 Nov 24, 2022
19 Dec 12, 2022
4 Aug 23, 2022
3 Jan 01, 2022
16 Sep 22, 2022
9 Dec 15, 2022