Tensorflow 2/1: faster GPU implementation & other small additions - Githubissues

roamanalytics / mittens

A fast implementation of GloVe, with optional retrofitting

Apache License 2.0

243 stars 31 forks source link

Tensorflow 2/1: faster GPU implementation & other small additions #16

Open andreiionutdamian opened 4 years ago

andreiionutdamian commented 4 years ago

Finalized in-GPU optimization without feed_dict for tf2 and tf1
preserved old memory-to-GPU via param no_feeds=False (default True)
added save
added saving embeddings each save_iters
added loss history plot save each save_iters