import argparse
from time import time
from math import ceil
# 3p
import torch
from tqdm import tqdm
# project
from utils.utils import create_dir_in_checkpoint
from utils.data_utils import DatasetLoader
from utils.logger import Logger
from utils.parser import Parser
from models.cycle_gan import ColorizationCycleGAN
parser = Parser()
args = parser.parse_args()
# Warning
if torch.cuda.is_available() and args.no_cuda:
print("WARNING: You have a CUDA device, so you should probably run with --cuda\nDo you want to continue without using GPU?")
response = str(input(" [YES to continue | ANY key to exit and restart the execution]: "))
if response != "YES":
sys.exit()
# Create Checkpoints dir
create_dir_in_checkpoint(args.name)
if __name__ == '__main__':
# Load dataset
