Cannot run the run command about flax-cifar

[jvlmdr]

Hey! Thanks for reporting this! It looks like the command in the readme is incorrect. Try using --config.arch instead of --config.model.arch:

python main.py --dataset_root=path/to/cifar10 --config=configs/default.py --config.arch=resnet_v1_18

[jvlmdr]

Also, to respond to your question, there are currently no pre-trained models; you need to train from scratch. (It doesn't take too long to train on CIFAR with one GPU.)

[HosseinMA96]

Thanks for the reply. Do you also know how to save the model?

[jvlmdr]

Check out the flax docs (https://flax.readthedocs.io/en/latest/guides/use_checkpointing.html). Feel free to implement this and make a pull request. Otherwise, I'll get around to it when I can.

I'm closing this specific issue as fixed by 6a0e0ba. Feel free to open a new issue for checkpointing / saved models!