Hi, I run the test.sh and something unexpected happened.

```
Traceback (most recent call last):
  File "test.py", line 130, in <module>
    assert opt.feat_dims == test_loader.get_feat_dims()
AssertionError
```
I did not make any change before this.