Skip to content

Commit

Permalink
Dynamically find logs directory (#8)
Browse files Browse the repository at this point in the history
Logs used to be fetched within the current execution directory. Fetch them in the module directory instead.
  • Loading branch information
CloudyOverhead authored Apr 24, 2021
1 parent e8a3a9c commit 015132f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions acrosome_counter/build_model.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,14 @@
# -*- coding: utf-8 -*-
"""Build the network."""

from os.path import join
from os.path import join, dirname, pardir

from detectron2 import model_zoo
from detectron2.config import get_cfg

from .load_dataset import QTY_CLASSES

LOG_DIR = join(".", "logs")
LOG_DIR = join(dirname(__file__), pardir, "logs")
PRETRAINED_CHECKPOINT = model_zoo.get_checkpoint_url(
"COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml"
)
Expand Down

0 comments on commit 015132f

Please sign in to comment.