use entry points instead of scripts in setup.py, potentially fixing windows issue as well as settings things up for multiprocessing

Author: lucidrains

setup.py

@@ -3,8 +3,12 @@
 setup(
   name = 'stylegan2_pytorch',
   packages = find_packages(),
-  scripts=['bin/stylegan2_pytorch'],
-  version = '0.21.1',
+  entry_points={
+      'console_scripts': [
+          'stylegan2_pytorch = stylegan2_pytorch.cli:main',
+      ],
+  },
+  version = '0.21.2',
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
   author = 'Phil Wang',

bin/stylegan2_pytorch renamed to stylegan2_pytorch/cli.py

@@ -102,5 +102,5 @@ def train_from_folder(
         if _ % 50 == 0:
             model.print_log()
 
-if __name__ == "__main__":
+def main():
     fire.Fire(train_from_folder)