From 0427f223d2c10dfd5bf9505494ad281166149f39 Mon Sep 17 00:00:00 2001 From: nv-nguyen Date: Tue, 16 May 2023 23:08:57 +0200 Subject: [PATCH] Remove wandb in network --- src/model/base_network.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/model/base_network.py b/src/model/base_network.py index db0894d..bf832ab 100644 --- a/src/model/base_network.py +++ b/src/model/base_network.py @@ -6,7 +6,6 @@ from src.utils.visualization_utils import put_image_to_grid from torchvision.utils import make_grid, save_image import os -import wandb import torchvision.transforms as transforms from src.model.loss import GeodesicError import multiprocessing