Conversation
|
This code works, but it is currently very slow. Need to very the op placement on different GPUs to figure out why it is slow. |
|
Ref #44 |
|
A few things:
It's probably cleaner to put this into the |
|
See also tensorflow/tensorflow#2126 |
|
I get the following error on this branch: InvalidArgumentError (see above for traceback): Cannot assign a device to node 'save/ShardedFilename_1': Could not satisfy explicit device specification '/device:GPU:1' because no supported kernel for GPU devices is available. |
|
@dennybritz : wanted to know if the branch is usable now. If there are some specific ToDos I can help with the implementation |
Allow the user to replicate the model on multiple GPUs. Still WIP and untested.