Skip to content

Commit

Permalink
Fix typo train_config to train_subset
Browse files Browse the repository at this point in the history
  • Loading branch information
harimkang committed Mar 13, 2024
1 parent b9ceb95 commit ff55d94
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -98,5 +98,5 @@ According to datasets, ``learning rate`` and ``batch size`` can be adjusted like
(otx) ...$ otx train --config <model_config_path> \
--data_root <path_to_data_root> \
--data.config.train_config.batch_size <batch_size_to_be_updated> \
--data.config.train_subset.batch_size <batch_size_to_be_updated> \
--optimizer.lr <learning_rate_to_be_updated>
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ For example, to decrease the batch size to 4, fix the number of epochs to 100, e

.. code-block:: shell
(otx) ...$ otx train ... --data.config.train_config.batch_size 4 \
(otx) ...$ otx train ... --data.config.train_subset.batch_size 4 \
--max_epochs 100
.. tab:: API
Expand All @@ -194,7 +194,7 @@ For example, to decrease the batch size to 4, fix the number of epochs to 100, e
from otx.core.data.module import OTXDataModule
from otx.engine import Engine
data_config = DataModuleConfig(..., train_config=SubsetConfig(..., batch_size=4))
data_config = DataModuleConfig(..., train_subset=SubsetConfig(..., batch_size=4))
datamodule = OTXDataModule(..., config=data_config)
engine = Engine(..., datamodule=datamodule)
Expand Down
4 changes: 2 additions & 2 deletions docs/source/guide/tutorials/base/how_to_train/detection.rst
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@ For example, to decrease the batch size to 4, fix the number of epochs to 100, e

.. code-block:: shell
(otx) ...$ otx train ... --data.config.train_config.batch_size 4 \
(otx) ...$ otx train ... --data.config.train_subset.batch_size 4 \
--max_epochs 100
.. tab:: API
Expand All @@ -276,7 +276,7 @@ For example, to decrease the batch size to 4, fix the number of epochs to 100, e
from otx.core.data.module import OTXDataModule
from otx.engine import Engine
data_config = DataModuleConfig(..., train_config=SubsetConfig(..., batch_size=4))
data_config = DataModuleConfig(..., train_subset=SubsetConfig(..., batch_size=4))
datamodule = OTXDataModule(..., config=data_config)
engine = Engine(..., datamodule=datamodule)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -275,7 +275,7 @@ For example, to decrease the batch size to 4, fix the number of epochs to 100 an

.. code-block:: shell
(otx) ...$ otx train ... --data.config.train_config.batch_size 4 \
(otx) ...$ otx train ... --data.config.train_subset.batch_size 4 \
--max_epochs 100
.. tab:: API
Expand All @@ -286,7 +286,7 @@ For example, to decrease the batch size to 4, fix the number of epochs to 100 an
from otx.core.data.module import OTXDataModule
from otx.engine import Engine
data_config = DataModuleConfig(..., train_config=SubsetConfig(..., batch_size=4))
data_config = DataModuleConfig(..., train_subset=SubsetConfig(..., batch_size=4))
datamodule = OTXDataModule(..., config=data_config)
engine = Engine(..., datamodule=datamodule)
Expand Down

0 comments on commit ff55d94

Please sign in to comment.