taming-transformers icon indicating copy to clipboard operation
taming-transformers copied to clipboard

where is the coco segmentation vqgan pre-train model

Open lainxx opened this issue 3 years ago • 2 comments

when generation segmentation to image i can't find this file "ckpt_path: logs/2020-11-07T00-08-54_cocostuffthings_vqvae_segmentation_bce/checkpoints/last.ckpt"

lainxx avatar Mar 24 '21 02:03 lainxx

It's included in the transformer checkpoint. If you want to extract it, run

python scripts/extract_submodel.py logs/2021-01-20T16-04-20_coco_transformer/checkpoints/last.ckpt coco_cond_stage.ckpt cond_stage_model

pesser avatar May 20 '21 22:05 pesser

When trying to train the transformer I found the following error: ModuleNotFoundError: No module named 'cutlit' If I change "culit" for "main": TypeError: cannot unpack non-iterable NoneType object

chinitaberrio avatar May 31 '21 06:05 chinitaberrio