fix bugs

Author: hh23333

README.md

@@ -42,7 +42,7 @@ python scripts/main.py --root PATH_TO_DATAROOT \
  --transforms random_flip\
  --optim sgd --lr 0.02\
  --stepsize 25 50\
- --staged-lr --height 384 --width 192\
+ --staged-lr --height 384 --width 128\
  --batch-size 32 --base-lr-mult 0.5
 ```
 ## To train PVPM

torchreid/losses/hard_mine_triplet_loss.py

@@ -4,7 +4,7 @@
 import torch
 import torch.nn as nn
 import numpy as np
-import miosqp
+# import miosqp
 
 class TripletLoss(nn.Module):
     """Triplet loss with hard positive/negative mining.