/GUPNet_For_Pycharm

Primary LanguagePythonMIT LicenseMIT

modify for pycharm running

  • code/experiments code/lib add init.py
  • modify code/experiments/config.yaml batchsize for 2080ti-12G
  • modify code/lib/helpers/train_helper.py
    • add os.environ["CUDA_VISIBLE_DEVICES"]="0"