libprotobuf FATAL : CHECK failed: it != end(): key not found

[gvanboven]

When trying to run train.py to train a model I get the following error:

[libprotobuf FATAL external/protobuf_archive/src/google/protobuf/map.h:1064] CHECK failed: it != end(): key not found: architecture terminate called after throwing an instance of 'google::protobuf::FatalException' what(): CHECK failed: it != end(): key not found: architecture Aborted (core dumped)

Any help on what the issue could be would be appreciated! I am using python 3.6 and CUDA 10.2