update readme
parent
e34e3ae85f
commit
4a0793780b
18
README.rst
18
README.rst
|
@ -97,14 +97,14 @@ Get started: 30 seconds to Torchreid
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
datamanager = torchreid.data.ImageDataManager(
|
datamanager = torchreid.data.ImageDataManager(
|
||||||
root='reid-data',
|
root="reid-data",
|
||||||
sources='market1501',
|
sources="market1501",
|
||||||
targets='market1501',
|
targets="market1501",
|
||||||
height=256,
|
height=256,
|
||||||
width=128,
|
width=128,
|
||||||
batch_size_train=32,
|
batch_size_train=32,
|
||||||
batch_size_test=100,
|
batch_size_test=100,
|
||||||
transforms=['random_flip', 'random_crop']
|
transforms=["random_flip", "random_crop"]
|
||||||
)
|
)
|
||||||
|
|
||||||
3 Build model, optimizer and lr_scheduler
|
3 Build model, optimizer and lr_scheduler
|
||||||
|
@ -112,9 +112,9 @@ Get started: 30 seconds to Torchreid
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
model = torchreid.models.build_model(
|
model = torchreid.models.build_model(
|
||||||
name='resnet50',
|
name="resnet50",
|
||||||
num_classes=datamanager.num_train_pids,
|
num_classes=datamanager.num_train_pids,
|
||||||
loss='softmax',
|
loss="softmax",
|
||||||
pretrained=True
|
pretrained=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -122,13 +122,13 @@ Get started: 30 seconds to Torchreid
|
||||||
|
|
||||||
optimizer = torchreid.optim.build_optimizer(
|
optimizer = torchreid.optim.build_optimizer(
|
||||||
model,
|
model,
|
||||||
optim='adam',
|
optim="adam",
|
||||||
lr=0.0003
|
lr=0.0003
|
||||||
)
|
)
|
||||||
|
|
||||||
scheduler = torchreid.optim.build_lr_scheduler(
|
scheduler = torchreid.optim.build_lr_scheduler(
|
||||||
optimizer,
|
optimizer,
|
||||||
lr_scheduler='single_step',
|
lr_scheduler="single_step",
|
||||||
stepsize=20
|
stepsize=20
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -149,7 +149,7 @@ Get started: 30 seconds to Torchreid
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
engine.run(
|
engine.run(
|
||||||
save_dir='log/resnet50',
|
save_dir="log/resnet50",
|
||||||
max_epoch=60,
|
max_epoch=60,
|
||||||
eval_freq=10,
|
eval_freq=10,
|
||||||
print_freq=10,
|
print_freq=10,
|
||||||
|
|
Loading…
Reference in New Issue