Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

README.md 1.7 KB

You have to be logged in to leave a comment. Sign In

Meta Pseudo Labels

Unofficial TF2 implementation of "Meta Pseudo Labels" (official Paper and Code).

Results

CIFAR-10-4K
w/o finetune w/ finetune
Paper 96.08%
Implementation 94.70% 94.71%

Usage

Setup Environment

python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
pre-commit install

Run Scripts

export PYTHONPATH=src

# download dataset
dvc pull -r origin

# train & evaluate
python src/main.py --data-dir data/cifar10 --config-name cifar10 --model-dir workdir/training --mpl-epochs 10000 --mpl-batch-size 64 --finetune-epochs 1000 --finetune-batch-size 512

# only evaluate
python src/evaluate.py --data-dir data/cifar10 --config-name cifar10 --saved-model-dir [workdir/training/finetune/model, saved_models/mpl, saved_models/finetune]
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...