Xuefeng Hu

Hi Johnny, To load data-parallel checkpoints on single GPU, you can try to make your model to be a dataparallel model first, load the model and change your model back....

My bad. It has been a while and I forgot that was tf code. For tensorflow you can check this code: https://stackoverflow.com/questions/41342098/keras-load-checkpoint-weights-hdf5-generated-by-multiple-gpus This is keras but I remember normal TF...

Dear David, You can try to load the model in multi-gpu model and then transform the model into single GPU. For tensorflow you can check this code: https://stackoverflow.com/questions/41342098/keras-load-checkpoint-weights-hdf5-generated-by-multiple-gpus This is...