Yuanhui Guo
Results
2
comments of
Yuanhui Guo
I modified some code of train.py, which can run on a single GPU(my GPU is RTX3070).Except modify train.py, you need modify model/model_stage.py : batchnorm2d (out_chan) modify to nn.BatchNorm2d(out_chan) train.py ```...
同问,我这边不使用这个model的话TensorFlow报模型使用错误