- 博客(3)
- 收藏
- 关注
转载 2020.5.26 docker-compose仍然无法使用 gpu属性
docker-compose版本: 3.X对于2.X版本,可以添加runtimes熟悉。参考:添加{ "runtimes": { "nvidia": { "path": "/usr/bin/nvidia-container-runtime", "runtimeArgs": [] } }}到 /etc/docker/daemon.json然后使用 --runtime=nvidiadocker-c
2020-05-26 16:04:03
833
原创 Keras 调用leaky_relu
Keras 中有leaky_relu的实现。leaky_relu被整合进了relu函数。参考官方文档:https://tensorflow.google.cn/api_docs/python/tf/keras/backend/relu?hl=enArgumentsxA tensor or variable.alphaA scalar, slope of negative section (default=0.).max_valuefloat. Saturati
2020-05-18 16:29:39
3717
原创 关于triplet loss的注意事项
triplet loss 的代码有一点需要注意:因为输出是对embedding层的映射后的距离的测度(一般是Euclidean距离)。标签变量(y)是辅助计算的变量。不能one-hot编码。借鉴下github上triplet loss的代码https://github.com/omoindrot/tensorflow-triplet-lossimport tensorflow as tfdef _pairwise_distances(embeddings, squared=False).
2020-05-12 15:14:09
729
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人