Commit c9c72937 authored by narugo1992's avatar narugo1992
Browse files

Merge branch 'ccip' of narugo.github.com:deepghs/imgutils into ccip

parents 09ddd40f 05b38663

zoo/ccip/train_.py

+1 −1
Original line number Diff line number Diff line
@@ -159,8 +159,8 @@ def train(dataset_dir: str, session_name: Optional[str] = None, from_ckpt: Optio 
        pred_list, gt_list = [], []

        model.train()

        num_iter = len(train_dataloader)

        for i, (inputs, char_ids) in enumerate(tqdm(train_dataloader)):

        train_dataloader.dataset.reset()

        for i, (inputs, char_ids) in enumerate(tqdm(train_dataloader)):

            inputs = inputs.to(accelerator.device)  # BxCxHxW

            char_ids = char_ids.to(accelerator.device)  # B