Skip to content

Instantly share code, notes, and snippets.

@adamjstewart

adamjstewart/train.py

Created May 5, 2022 21:42
Show Gist options
  • Save adamjstewart/3feabfc4a54a7126f1ef189f02da7ffa to your computer and use it in GitHub Desktop.
Save adamjstewart/3feabfc4a54a7126f1ef189f02da7ffa to your computer and use it in GitHub Desktop.
Download ZIP
Training a model with TorchGeo
Raw
train.py
for batch in dataloader:
image = batch["image"]
mask = batch["mask"]
# train a model, or make predictions using a pre-trained model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment