def test_case_evaluate_FID_IS_ddp(self): """ Usage: export PYTHONWARNINGS=ignore export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 export TIME_STR=1 export PYTHONPATH=./exp:./stylegan2-pytorch:./ python -c "from template_lib.v2.tests.test_GAN import TestingTFFIDISScore;\ TestingTFFIDISScore().test_case_evaluate_FID_IS_ddp()" :return: """ if 'CUDA_VISIBLE_DEVICES' not in os.environ: os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' if 'TIME_STR' not in os.environ: os.environ['TIME_STR'] = '0' if utils.is_debugging() else '0' command, outdir = get_command_and_outdir( self, func_name=sys._getframe().f_code.co_name, file=__file__) argv_str = f""" --tl_config_file none --tl_command none --tl_outdir {outdir} """ args = setup_outdir_and_yaml(argv_str) nproc_per_node = len(os.environ['CUDA_VISIBLE_DEVICES'].split(',')) cmd_str = f""" python -m torch.distributed.launch --nproc_per_node={nproc_per_node} --master_port=8888 template_lib/v2/GAN/evaluation/tf_FID_IS_score.py --run_func TFFIDISScore.test_case_evaluate_FID_IS """ cmd_str += get_append_cmd_str(args) start_cmd_run(cmd_str) pass
def test_test_vae(self): """ Usage: export CUDA_VISIBLE_DEVICES=1 export TIME_STR=1 export PYTHONPATH=./PyTorch_VAE_lib:./ python -c "from exp.tests.test_pytorch_vae import Testing_PyTorch_VAE;\ Testing_PyTorch_VAE().test_test_vae()" :return: """ if 'CUDA_VISIBLE_DEVICES' not in os.environ: os.environ['CUDA_VISIBLE_DEVICES'] = '1' if 'TIME_STR' not in os.environ: os.environ['TIME_STR'] = '0' if utils.is_debugging() else '1' command, outdir = get_command_and_outdir( self, func_name=sys._getframe().f_code.co_name, file=__file__) argv_str = f""" --tl_config_file exp/configs/pytorch_vae.yaml --tl_command {command} --tl_outdir {outdir} """ args = setup_outdir_and_yaml(argv_str) nproc_per_node = len(os.environ['CUDA_VISIBLE_DEVICES'].split(',')) cmd_str = f""" python exp/scripts/test.py -c PyTorch_VAE_lib/configs/vae.yaml """ cmd_str += get_append_cmd_str(args) start_cmd_run(cmd_str) pass