search

wzzheng / DCML

Codes for our CVPR 2021 paper "Deep Compositional Metric Learning"
MIT License
19 stars 3 forks source link

AttributeError: module 'batchminer' has no attribute 'BATCHMINING_METHODS' #1

Open JonasZero opened 3 years ago

JonasZero commented 3 years ago

This error occurs when I run the following code： python main.py --dataset cub200 --tau 55 --gamma 0.2 --gpu 0 --seed 0 --compos_num 4 --ensemble_num 4 --embed_dim 128 --bs 100 --n_epochs 300 --samples_per_class 2 --loss margin --batch_mining distance --arch resent50_frozen_normalize

Traceback (most recent call last): File "main.py", line 108, in import criteria as criteria File "/home/disk/github/DCML-main/criteria/init.py", line 1, in from criteria import triplet, margin, proxynca, npair File "/home/disk/github/DCML-main/criteria/triplet.py", line 6, in ALLOWED_MINING_OPS = list(batchminer.BATCHMINING_METHODS.keys()) AttributeError: module 'batchminer' has no attribute 'BATCHMINING_METHODS'

wangck20 commented 3 years ago

Sorry for late reply. We have made the correction of the code. The error will not occur again. Thanks for your question.