Skip to content

Commit

Permalink
Merge pull request #72 from plantnet/dev
Browse files Browse the repository at this point in the history
v2.1.1
  • Loading branch information
tlarcher authored Nov 14, 2024
2 parents cc37eb1 + 7662e16 commit 4687bf1
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -85,10 +85,10 @@ def __init__(
loss_kwargs['pos_weight'] = Tensor([loss_kwargs['pos_weight']] * num_classes)
super().__init__(model, optimizer=optimizer, metrics=metrics, task=task, loss_kwargs=loss_kwargs, hparams_preprocess=hparams_preprocess, checkpoint_path=checkpoint_path)
if self.model.pretrained and not self.checkpoint_path:
self.download_weights("https://lab.plantnet.org/seafile/f/d780d4ab7f6b419194f9/?dl=1",
self.download_weights("https://lab.plantnet.org/seafile/f/755f49a508db4552b7db/?dl=1",
weights_dir,
filename="pretrained.ckpt",
md5="69111dd8013fcd8e8f4504def774f3a5")
md5="3c2c17d0781016cf71c231761ae73dcc")

def forward(self, x, y, z): # noqa: D102 pylint: disable=C0116
return self.model(x, y, z)
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
from setuptools import find_packages, setup

setup(name="malpolon",
version="2.1.0",
description="Malpolon v2.1.0",
version="2.1.1",
description="Malpolon v2.1.1",
author="Theo Larcher, Titouan Lorieul, Benjamin Deneu, Lukas Picek",
author_email="[email protected], [email protected], [email protected], [email protected]",
url="https://github.com/plantnet/malpolon",
Expand Down

0 comments on commit 4687bf1

Please sign in to comment.