check for current directory

new_project.py

@@ -6,8 +6,13 @@
 # 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 len(sys.argv) == 2, 'Specify a name for the new project. Example: python3 new_project.py MyNewProject'
+
 project_name = Path(sys.argv[1])
+target_dir = current_dir / project_name
 
 ignore = [".git", "data", "saved", "new_project.py", "LICENSE", ".flake8", "README.md", "__pycache__"]
-copytree('.', project_name, ignore=ignore_patterns(*ignore))
+copytree(current_dir, target_dir, ignore=ignore_patterns(*ignore))
+print('New project initialized at', target_dir.absolute().resolve())