* Fix a bug when load checkpoints in mulit-GPUs environment. * Map the weights to CPU |
||
---|---|---|
.. | ||
__init__.py | ||
inference.py | ||
test.py | ||
train.py |
* Fix a bug when load checkpoints in mulit-GPUs environment. * Map the weights to CPU |
||
---|---|---|
.. | ||
__init__.py | ||
inference.py | ||
test.py | ||
train.py |