Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Apr 4, 2023
1 parent de4c51f commit 42db92b
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 9 deletions.
8 changes: 2 additions & 6 deletions setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -16,10 +16,6 @@ classifiers =
Programming Language :: Python
Programming Language :: Python :: 3
Programming Language :: Python :: 3 :: Only
Programming Language :: Python :: 3.6
Programming Language :: Python :: 3.7
Programming Language :: Python :: 3.8
Programming Language :: Python :: 3.9
Programming Language :: Python :: Implementation :: CPython

[options]
Expand All @@ -29,10 +25,10 @@ install_requires =
fast-overlap
numpy
pandas
read_roi
read-roi
tifffile
torch
python_requires = >=3.6
python_requires = >=3.7

[options.extras_require]
test =
Expand Down
2 changes: 1 addition & 1 deletion yeast_mrcnn/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def mrcnn():
)

# Make anchor generator with 3 sizes per feature map and 5 aspect ratios
sizes = tuple(2.0 ** x for x in range(5, 12))
sizes = tuple(2.0**x for x in range(5, 12))
aspects = tuple(0.5 * x for x in range(1, 5))
n_feature_maps = 5 # true for resnet50 with FPN
ag_sizes = tuple(tuple(sizes[i : i + 3]) for i in range(n_feature_maps))
Expand Down
2 changes: 0 additions & 2 deletions yeast_mrcnn/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@


def train_one_epoch(model, dataloader, optimizer, epoch, device):

loss_df = pd.DataFrame()

lr_scheduler = None
Expand Down Expand Up @@ -70,7 +69,6 @@ def train(
model.train()

for e in range(epochs):

loss_df = train_one_epoch(model, train_dataloader, optimizer, e, device)

loss_df = loss_df.join(evaluate_test(model, val_dataloader, device))
Expand Down

0 comments on commit 42db92b

Please sign in to comment.