adamjstewart/train.py

Created May 5, 2022 21:42

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/adamjstewart/3feabfc4a54a7126f1ef189f02da7ffa.js"></script>
Save adamjstewart/3feabfc4a54a7126f1ef189f02da7ffa to your computer and use it in GitHub Desktop.

Download ZIP

Training a model with TorchGeo

Raw

	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