fix bugs for dataset initialization.#436

Golden-Pigeon wants to merge 1 commit intosczhou:mastersczhou/CodeFormer:masterfrom

Golden-Pigeon:masterGolden-Pigeon/CodeFormer:masterCopy head branch name to clipboard