Skip to content

TF2 compatible and pypi publish#34

Open
chnzhangrui wants to merge 13 commits intovlimant:masterfrom
chnzhangrui:master
Open

TF2 compatible and pypi publish#34
chnzhangrui wants to merge 13 commits intovlimant:masterfrom
chnzhangrui:master

Conversation

@chnzhangrui
Copy link

@chnzhangrui chnzhangrui commented Jul 30, 2020

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