tf
zz__2020
这个作者很懒,什么都没留下…
展开
-
tfa.seq2seq.LuongAttention解析
源码class LuongAttention(AttentionMechanism): """Implements Luong-style (multiplicative) attention scoring. This attention has two forms. The first is standard Luong attention, as described in: Minh-Thang Luong, Hieu Pham, Christopher D. Ma原创 2021-02-28 16:48:56 · 614 阅读 · 0 评论 -
tfa.seq2seq.BasicDecoder的源码解读
源码获取https://www.tensorflow.org/addons/api_docs/python/tfa/seq2seq/BasicDecodergithub上的源码class BasicDecoder(decoder.BaseDecoder): """Basic sampling decoder for training and inference. The `tfa.seq2seq.Sampler` instance passed as argument is resp原创 2021-02-26 11:21:05 · 653 阅读 · 1 评论 -
多个gpu指定
代码import osos.environ["CUDA_VISIBLE_DEVICES"] = "3,2,0,1"结果实际显卡编号----->运算显卡编号 3 -----> 0(主卡) 2 -----> 1 0 -----> 2 1 -----> 3原创 2021-02-25 21:58:28 · 423 阅读 · 0 评论 -
不同tf对应python以及cuda cudnn的版本
tf官网查询https://www.tensorflow.org/install/source#linux目前的对应关系原创 2021-02-24 21:30:15 · 1362 阅读 · 0 评论