search

CaptainEven / FaceRecognition

Face recognition using triplet loss, implementing FaceNet with pytorch.人脸识别项目，提供一个小型数据集用作验证，使用三元组损失函数(Triplet loss)提升准确率和泛化能力，对FaceNet进行了一种实现。
127 stars 39 forks source link

pytorch vision #5

Open wangshuo6699 opened 5 years ago

wangshuo6699 commented 5 years ago

请问，使用的是什么版本的pytorch？

little2Rabbit commented 5 years ago

请问下，您用什么版本的pytorch和torchision

CaptainEven commented 5 years ago

@wangshuo6699 @qq542288564 pytorch0.3.1吧，改成pytorch1.0及以上很容易，去掉Variable即可。

zhilangtaosha commented 5 years ago

改成pytorch1.0及以上很容易，去掉Variable即可。 没看明白是什么意思，能否详细解释一下？