Skip to content

The model obtained after fine-tuning, how can it be used for inference? #308

@RuipingWang1986

Description

@RuipingWang1986

Hello, I have fine-tuned the pretrained model imagenet21k_ViT-B_16.npz using the command python -m vit_jax.main --workdir=/tmp/vit-$(date +%s) \ --config=$(pwd)/vit_jax/configs/vit.py:b16,cifar10 \ --config.pretrained_dir='gs://vit_models/imagenet21k', and the fine-tuned results have been saved to a folder named vit-44527685. However, there is no .npz file in this folder, and I also don’t know how to use the model I obtained after fine-tuning for inference services to verify the effectiveness of the fine-tuning. Could you please help me figure out where the problem lies?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions