alibaba / ai-matrix

To make it easy to benchmark AI accelerators
Other
179 stars 79 forks source link

module 'mlperf_compliance.mlperf_log' has no attribute 'setdefault' in SSD_ResNet34_PyTorch #68

Open houseofai opened 4 years ago

houseofai commented 4 years ago

Hi there, I'm having facing an issue when trying to training SSD_ResNet34_PyTorch.

Inside macro_benchmark/SSD_ResNet34_PyTorch, if I run the default command: python -u train.py --local_rank=0 --use-fp16 --nhwc --pad-input --jit --delay-allreduce --opt-loss --epochs 10 --batch-size 128 --max_iter 3200 --warmup-factor 0 --no-save

I get the following error:

Traceback (most recent call last): File "train.py", line 857, in <module> main() File "train.py", line 830, in main mlperf_compliance.mlperf_log.setdefault( AttributeError: module 'mlperf_compliance.mlperf_log' has no attribute 'setdefault'

I run that code inside the recommended docker image: nvcr.io/nvidia/pytorch:19.05-py3

Looking at mlperf_compliance lib, I don't see any 'setdefault' method.