testchujtest

TypeError: __init__() got an unexpected keyword argument 'roi_head'

2

Hi, i tried to train this model on custom data. I built docker image: ``` ARG PYTORCH="1.4" ARG CUDA="10.1" ARG CUDNN="7" FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0+PTX" ENV TORCH_NVCC_FLAGS="-Xfatbin -compress-all"...