caffe-cpu

public

10 stars

3 forks

1 issues

Commits

List of commits on branch master.

Unverified

2cefaeaf3f4943ba8835f6c9462e57dae7f9b897

added project report, simple Readme and TODO

committed 9 years ago

Unverified

b0964df8f8009ba6acfcea00c3ec511960b5825d

Some changes to parameters

jjinala committed 9 years ago

Unverified

0efbe63410c8ae0e099033af05537da971ea42d0

Added noise

jjinala committed 9 years ago

Unverified

afd4c43792408f59b54b116e31e6fff9e8cf55f8

Added a distance metric layer and scripts to generate data for the grid example.

jjinala committed 9 years ago

Unverified

5f4389707c03c4570837bfac3efa298e28cee9ce

Some parameter changes.

jjinala committed 9 years ago

Unverified

412f09ee82c91f0d860a353631e76abe5f40513f

Fix to initialization of distance matrix.

jjinala committed 9 years ago

README

The README file for this repository.

Wasserstein Loss Layer

Implements the Wasserstein Loss Layer (as described in Frogner et al. 2015) in Caffe. (Previously, the only open source implementation of the loss function was in Mocha.jl, which we used as a reference)

Project Report

Check ./project.pdf

Example Models using the loss layer

Check ./examples/mnist and ./examples/sample_wasserstein

TODO

Fix the test to check for difference in gradient.