Skip to content

TF2 compatible and pypi publish - #34

Open
chnzhangrui wants to merge 13 commits into
vlimant:masterfrom
chnzhangrui:master
Open

chnzhangrui wants to merge 13 commits into
vlimant:masterfrom
chnzhangrui:master

Conversation

@chnzhangrui

@chnzhangrui chnzhangrui commented Jul 30, 2020

Copy link
Copy Markdown

It works with TF2 now without explicit Keras.
I also made a channel for it https://pypi.org/project/nnlo/0.0.4/.
MNIST example now runs with

pip install nnlo
GetData MNIST
jsrun -n 12 -g 1 TrainingDriver --model mnist --loss categorical_crossentropy --epochs 1
jsrun -n 12 -g 1 TrainingDriver --model mnist_torch --loss categorical_crossentropy --epochs 1

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant