8000 About batch_ size · Issue #4 · joeyz0z/ConZIC · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
About batch_ size #4
Open
Open
@Rouchestand

Description

@Rouchestand

Set caption_img_path to ./examples and batch_size to 2. Running run.py will result in the following error:
Traceback (most recent call last):
File "run.py", line 162, in
run_control(run_type, args, img_dir, lm_model, lm_tokenizer, clip, token_mask)
File "run.py", line 87, in run_control
gen_texts, clip_scores = control_generate_caption(lm_model, clip, lm_tokenizer, image_instance, token_mask,
File "/home/bishe/prc/ConZIC-main/ConZIC-main/control_gen_utils.py", line 200, in control_generate_caption
generate_texts, clip_scores = sentiment_sequential_generation(model, clip, tokenizer, image_instance, token_mask, prompt,
File "/home/bishe/prc/ConZIC-main/ConZIC-main/control_gen_utils.py", line 54, in sentiment_sequential_generation
topk_inp[:, ii + seed_len] = idxs_
RuntimeError: The expanded size of the tensor (400) must match the existing size (200) at non-singleton dimension 0. Target sizes: [400]. Tensor sizes: [200]

< 4CB1 /div>

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions

      0