Skip to content

Commit

Permalink
add hint for weights
Browse files Browse the repository at this point in the history
dketterer committed Aug 16, 2019
1 parent 8b97aa9 commit 810d29a
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -2,6 +2,8 @@

Code to train and run a Colorization U-Net.

**Trained weights can be found in the releases**

# Prerequisites

Machine needs a Nvidia GPU with installed drivers and CUDA 10.0 (for Tensorflow 1.14).

0 comments on commit 810d29a

Please sign in to comment.