Problem with training model ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8

I have prepared all documents for training this model but I have an error in details:
TypeError: init(): incompatible constructor arguments. The following argument types are supported:
1. tensorflow.python.lib.io._pywrap_file_io.BufferedInputStream(filename: str, buffer_size: int, token: tensorflow.python.lib.io._pywrap_file_io.TransactionToken = None)

Invoked with: None, 524288
I need some help for my project.
*Note: I train this model in VS Code