Official Pytorch implementation of Meta Internal Learning

Last update: Aug 24, 2022

Related tags

Overview

Meta Internal Learning

This repository is the official implementation of Meta Internal Learning by Raphael Bensadoun, Shir Gur, Tomer Galanti, Lior Wolf.

Project | arXiv | Code

Requirements

To install requirements:

pip install -r requirements.txt

Training on a dataset

Create a folder X containing the images. (see structure in data folder)
Determine how many iterations Y to train by scale (depends mostly on the size of the dataset, you may refer to the appendix for reference).
Run

python train.py --image-path X --batch-size 16 --visualize --niter Y --min-size 25 --checkname X_result --SAVE-MODEL --SAVE-IMGS

Generated images, tensorboard logs and trained models are stored in MetaInternalLearning/run/X_result.

The default input format is jpg. Use '--file-suffix png' for .png files.

Examples from paper -

Places-50 -

python train.py --image-path data/places_50 --batch-size 16 --visualize --niter 4000  --min-size 28 --checkname places_50_result --SAVE-MODEL --SAVE-IMGS

LSUN-50

python train.py --image-path data/lsun_50 --batch-size 16 --visualize --niter 5000  --checkname lsun_50_result --SAVE-MODEL --SAVE-IMGS

Valley dataset can be downloaded here - http://places2.csail.mit.edu/download.html (256x256 small images) and can be divided into subsets as mentioned in the paper.

V500 -

python train.py --image-path data/V500 --batch-size 16 --visualize --niter 25000 --min-size 25 --checkname v500_result --ar 1 --SAVE-MODEL

V2500 -

python train_dataset_parallel.py --image-path data/V2500 --batch-size 16 --niter 100000 --rec-weight 50 --min-size 25 --checkname v2500_result --ar 1 --SAVE-MODEL

V5000 -

python train_dataset_parallel.py --image-path data/V5000 --batch-size 16 --niter 150000 --rec-weight 50 --min-size 25 --checkname v5000_result --ar 1 --SAVE-MODEL

Applications

Coming soon!

Official Pytorch implementation of Meta Internal Learning

Related tags

Overview

Meta Internal Learning

Requirements

Training on a dataset

Examples from paper -

Applications

Owner

This code reproduces the results of the paper, "Measuring Data Leakage in Machine-Learning Models with Fisher Information"

True Few-Shot Learning with Language Models

WormMovementSimulation - 3D Simulation of Worm Body Movement with Neurons attached to its body

Graph Regularized Residual Subspace Clustering Network for hyperspectral image clustering

Official implementation of Rethinking Graph Neural Architecture Search from Message-passing (CVPR2021)

A resource for learning about deep learning techniques from regression to LSTM and Reinforcement Learning using financial data and the fitness functions of algorithmic trading

Source code for our paper "Improving Empathetic Response Generation by Recognizing Emotion Cause in Conversations"

Object Detection using YOLO from PyImageSearch

R3Det based on mmdet 2.19.0

Direct design of biquad filter cascades with deep learning by sampling random polynomials.

CFNet: Cascade and Fused Cost Volume for Robust Stereo Matching（CVPR2021）

An attempt at the implementation of GLOM, Geoffrey Hinton's paper for emergent part-whole hierarchies from data

AbelNN: Deep Learning Python module from scratch

The official PyTorch implementation of Curriculum by Smoothing (NeurIPS 2020, Spotlight).

Selective Wavelet Attention Learning for Single Image Deraining

Feup-csr - Repository holding my group's submission to the CSR project competition

for a paper about leveraging discourse markers for training new models

This framework implements the data poisoning method found in the paper Adversarial Examples Make Strong Poisons

Subdivision-based Mesh Convolutional Networks

Inteligência artificial criada para realizar interação social com idosos.