new_project.py

import sys
from pathlib import Path
from shutil import copytree, ignore_patterns


# This script initializes new pytorch project with the template files.
# Run `python3 new_project.py ../MyNewProject` then new project named
# MyNewProject will be made
current_dir = Path()
assert (current_dir / 'new_project.py').is_file(
), 'Script should be executed in the pytorch-template directory'
assert_msg = "Specify a name for the new project. "
"Example: python3 new_project.py MyNewProject"
assert len(sys.argv) == 2, assert_msg

project_name = Path(sys.argv[1])
target_dir = current_dir / project_name

ignore = ["downloaded-data", "saved", "new_project.py",
          "LICENSE", ".flake8", "README.md", "__pycache__"]
copytree(current_dir, target_dir, ignore=ignore_patterns(*ignore))
print('New project initialized at', target_dir.absolute().resolve())