search

google-research / big_vision

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.
Apache License 2.0
2.04k stars 140 forks source link

Errors in notebooks #93

Open amrzv opened 5 months ago

amrzv commented 5 months ago

Hi.

  1. When running notebook in colab uvim_depth_task.ipynb on line oracle_params, oracle_state = vit.load(None, "depth_stageI_params.npz") the error is raised

    AttributeError: module 'big_vision.utils' has no attribute 'load_checkpoint'

    image

  2. The same error in clippo_colab.ipynb on line

    params = utils.load_checkpoint(None, checkpoint_path)['params']

    image

  3. When running lit.ipynb on line

    params0 = model.init(jax.random.PRNGKey(42), *init_params)['params'].unfreeze()

    the error is raised

    AttributeError: 'dict' object has no attribute 'unfreeze'

    image