Closed devmessias closed 5 years ago
Aparentemente existe um bug relacionado com o paralellismo do tf.map_fn em GPU https://github.com/tensorflow/tensorflow/issues/14623 https://github.com/tensorflow/tensorflow/issues/24774 Enquanto em CPU eu consigo o paralelismo em GPU não
Uma possibilidade para contornar o bug do paralelismo é usar operações matriciais para defini-lo, ao invés de tf.map_fn
See 819e84c4d295ffd1044994fa75d16