Ejemplos de rm_dir en Python

Ejemplo n.º 1

Archivo: retrieve_images.py Proyecto: fortierq/deep-learning-for-multimodal-image-retrieval

def save_image_vectors(model, dataloader_test):
    utils.rm_dir(Dir.image_vectors)
    with torch.no_grad():
        model.eval()
        for img, _, img_name, _ in dataloader_test:
            vectors = model(img.to("cuda"))
            for i, v in enumerate(vectors):
                path = Dir.image_vectors / Path(img_name[i]).relative_to(Dir.images)
                path.parent.mkdir(exist_ok=True, parents=True)
                torch.save(v, str(path))

Ejemplo n.º 2

 def _create_terraform_root(self):
     utils.rm_dir(self.terraform_root)
     utils.mkdir_p(self.terraform_root)

Ejemplo n.º 3

 def checkpoint_flush(self, silent=True):
     rm_dir(self.ckpt_dir, silent)

Ejemplo n.º 4

 def summary_flush(self, silent=True):
     rm_dir(self.summary_dir, silent)