Skip to content

Initial model for v2#349

Open
alejandromarcu wants to merge 3 commits intomainfrom
initial_model
Open

Initial model for v2#349
alejandromarcu wants to merge 3 commits intomainfrom
initial_model

Conversation

@alejandromarcu
Copy link
Collaborator

This allows to start the training from a model stored in either wandb or a local file.
E.g.:

training:
  initial_model:
    wandb_project: B5W3
    wandb_alias: v160

@adamantivm
Copy link
Collaborator

One question: does this also pull the model from wandb and store it as onnx so that self-play can work with rust?

@alejandromarcu
Copy link
Collaborator Author

One question: does this also pull the model from wandb and store it as onnx so that self-play can work with rust?

Yes. The model is stored in wandb just as a pt, but when you specify the initial model, we create the AlphaZero agent with that model, and then we save it as model_0.pt and model_0.onnx if save_onnx=True.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants