Python torch.utils.collect_env.get_pretty_env_info() Examples

The following are 27 code examples of torch.utils.collect_env.get_pretty_env_info(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module torch.utils.collect_env , or try the search function

Example #1

Source File: trainer.py From elektronn3 with MIT License

6 votes

def archive_backup(self):
        """Archiving the source folder, the training script and environment info.

        The training script is saved with the prefix "0-" to distinguish from regular scripts.
        Environment information equivalent to the output of ``python -m torch.utils.collect_env``
        is saved in a file named "env_info.txt".
        """

        # Archiving the Training script
        shutil.copyfile(self.script_path, self.save_path + '/0-' + os.path.basename(self.script_path))
        os.chmod(self.save_path + '/0-' + os.path.basename(self.script_path), 0o755)
        # Archiving the src folder
        pkg_path = os.path.dirname(arch_src)
        backup_path = os.path.join(self.save_path, 'src_backup')
        shutil.make_archive(backup_path, 'gztar', pkg_path)

        # Archiving the Environment Info
        env_info = collect_env.get_pretty_env_info()
        with open(self.save_path + '/env_info.txt', 'w') as f:
            f.write(env_info)

Example #2

Source File: collect_env.py From Det3D with Apache License 2.0

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #3

Source File: collect_env.py From Res2Net-maskrcnn with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #4

Source File: collect_env.py From TinyBenchmark with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #5

Source File: collect_env.py From maskscoring_rcnn with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #6

Source File: collect_env.py From DF-Traffic-Sign-Identification with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #7

Source File: collect_env.py From RRPN_pytorch with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #8

Source File: collect_env.py From NAS-FCOS with BSD 2-Clause "Simplified" License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #9

Source File: collect_env.py From training with Apache License 2.0

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #10

Source File: collect_env.py From detectron2 with Apache License 2.0

5 votes

def collect_torch_env():
    try:
        import torch.__config__

        return torch.__config__.show()
    except ImportError:
        # compatible with older versions of pytorch
        from torch.utils.collect_env import get_pretty_env_info

        return get_pretty_env_info()

Example #11

Source File: train_siammask_refine.py From SiamMask with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += "\n        OpenCV ({})".format(cv2.__version__)
    return env_str

Example #12

Source File: train_siammask.py From SiamMask with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += "\n        OpenCV ({})".format(cv2.__version__)
    return env_str

Example #13

Source File: train_siamrpn.py From SiamMask with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += "\n        OpenCV ({})".format(cv2.__version__)
    return env_str

Example #14

Source File: collect_env.py From detectron2 with Apache License 2.0

5 votes

def collect_torch_env():
    try:
        import torch.__config__

        return torch.__config__.show()
    except ImportError:
        # compatible with older versions of pytorch
        from torch.utils.collect_env import get_pretty_env_info

        return get_pretty_env_info()

Example #15

Source File: collect_env.py From retinamask with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #16

Source File: osutils.py From Relation-Aware-Global-Attention-Networks with MIT License

5 votes

def collect_env_info():
    """Returns env info as a string.
    Code source: github.com/facebookresearch/maskrcnn-benchmark
    """
    from torch.utils.collect_env import get_pretty_env_info
    env_str = get_pretty_env_info()
    env_str += '\n        Pillow ({})'.format(PIL.__version__)
    return env_str

Example #17

Source File: collect_env.py From FreeAnchor with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #18

Source File: collect_env.py From EmbedMask with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #19

Source File: collect_env.py From maskrcnn-benchmark with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #20

Source File: tools.py From deep-person-reid with MIT License

5 votes

def collect_env_info():
    """Returns env info as a string.

    Code source: github.com/facebookresearch/maskrcnn-benchmark
    """
    from torch.utils.collect_env import get_pretty_env_info
    env_str = get_pretty_env_info()
    env_str += '\n        Pillow ({})'.format(PIL.__version__)
    return env_str

Example #21

Source File: collect_env.py From HRNet-MaskRCNN-Benchmark with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #22

Source File: collect_env.py From sampling-free with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #23

Source File: collect_env.py From remote_sensing_object_detection_2019 with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #24

Source File: collect_env.py From DetNAS with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #25

Source File: collect_env.py From Clothing-Detection with GNU General Public License v3.0

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #26

Source File: collect_env.py From DenseMatchingBenchmark with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str

Example #27

Source File: collect_env.py From R2CNN.pytorch with MIT License

5 votes

def collect_env_info():
    env_str = get_pretty_env_info()
    env_str += get_pil_version()
    return env_str