Skip to content

sliorde/generating-neural-networks-with-neural-networks

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

generating-neural-networks-with-neural-networks

This repository contains a TensorFlow implementation of a hypernetwork, as described here. Also contained in the repo is the code required to reproduce results from the paper.

The hypernetwork model is implemented in hypernetwork.py. To train it, run train.py. It uses the hyper-parameters defined in params.py, which are the same parameters used in the paper.

The folder MNF has an implementation of MNFG . The implementation was cloned from here, but it was modified so that it can be used in my experiments. I regret to say that my modifications are very messy, which might make it hard to use this code. I suggest that you go to the original repo, use that code to train the network, save the checkpoints in some folder, and then use my code to run experiments with these checkpoints.

The folder analysis has scripts required to run the experiments described in the paper. The scripts whose file name ends with _data_creator are used to create data, which can then be displayed using the other scripts, whose file names end with _display . Each script mentions in its body which data it requires. The name of these scripts (before the _display suffix) matches the title of the subsection in the paper in which the corresponding experiment was described.

Finally, the folder toy_problem has the toy example from the paper. To train the model, run toy_example.py.

About

Code that goes along with paper http://arxiv.org/abs/1801.01952

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages