search

clcarwin / sphereface_pytorch

A PyTorch Implementation of SphereFace.
MIT License
714 stars 172 forks source link

'float' object cannot be interpreted as an integer #39

Closed leadcain84 closed 6 years ago

leadcain84 commented 6 years ago

Anyone solve same problem?

File "train.py", line 142, in train(epoch,args) File "train.py", line 101, in train bs=args.bs,shuffle=True,nthread=6,imagesize=128) File "/DL/pytorch/sphereface_pytorch/dataset.py", line 60, in init self.nparrimage = np.frombuffer(self.arrimage.get_obj(),np.float32).reshape(10,len(self.arrimage)/10) TypeError: 'float' object cannot be interpreted as an integer

leadcain84 commented 6 years ago

I solved it. change the "(self.arrimage)/10)" to "(self.arrimage)//10)"

my env is follow: pytorch 0.2.0 python 3.4