diff --git a/projects/FastAttr/fastattr/__init__.py b/projects/FastAttr/fastattr/__init__.py index fe20323..e1b8997 100644 --- a/projects/FastAttr/fastattr/__init__.py +++ b/projects/FastAttr/fastattr/__init__.py @@ -6,7 +6,6 @@ from .attr_evaluation import AttrEvaluator from .config import add_attr_config -from .data_build import build_attr_train_loader, build_attr_test_loader from .datasets import * from .modeling import * from .attr_dataset import AttrDataset diff --git a/projects/FastRetri/configs/cars.yml b/projects/FastRetri/configs/cars.yml index 2edbea3..4c80b53 100644 --- a/projects/FastRetri/configs/cars.yml +++ b/projects/FastRetri/configs/cars.yml @@ -22,6 +22,7 @@ SOLVER: BASE_LR: 0.05 ETA_MIN_LR: 0.0005 + NESTEROV: False MOMENTUM: 0. TEST: diff --git a/projects/FastRetri/configs/cub.yml b/projects/FastRetri/configs/cub.yml index 8b1470c..2d0cdf7 100644 --- a/projects/FastRetri/configs/cub.yml +++ b/projects/FastRetri/configs/cub.yml @@ -22,6 +22,7 @@ SOLVER: BASE_LR: 0.02 ETA_MIN_LR: 0.00002 + NESTEROV: False MOMENTUM: 0. TEST: