Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenPCDet
Commits
44c10a3a
"git@developer.sourcefind.cn:kecinstone/2024-pra-vllm.git" did not exist on "d4c7755ca88aefe749346f510be1b919186be0b6"
Commit
44c10a3a
authored
Jul 30, 2020
by
Shaoshuai Shi
Browse files
sort python imports with isort
parent
2b539530
Changes
68
Show whitespace changes
Inline
Side-by-side
Showing
8 changed files
with
37 additions
and
27 deletions
+37
-27
tools/eval_utils/eval_utils.py
tools/eval_utils/eval_utils.py
+5
-3
tools/test.py
tools/test.py
+10
-9
tools/train.py
tools/train.py
+10
-8
tools/train_utils/optimization/__init__.py
tools/train_utils/optimization/__init__.py
+4
-2
tools/train_utils/optimization/fastai_optim.py
tools/train_utils/optimization/fastai_optim.py
+1
-1
tools/train_utils/optimization/learning_schedules_fastai.py
tools/train_utils/optimization/learning_schedules_fastai.py
+3
-1
tools/train_utils/train_utils.py
tools/train_utils/train_utils.py
+3
-2
tools/visual_utils/visualize_utils.py
tools/visual_utils/visualize_utils.py
+1
-1
No files found.
tools/eval_utils/eval_utils.py
View file @
44c10a3a
import
tqdm
import
time
import
pickle
import
time
import
numpy
as
np
import
torch
from
pcdet.utils
import
common_utils
import
tqdm
from
pcdet.models
import
load_data_to_gpu
from
pcdet.utils
import
common_utils
def
statistics_info
(
cfg
,
ret_dict
,
metric
,
disp_dict
):
...
...
tools/test.py
View file @
44c10a3a
import
argparse
import
datetime
import
glob
import
os
import
re
import
time
from
pathlib
import
Path
import
numpy
as
np
import
torch
from
tensorboardX
import
SummaryWriter
import
time
import
glob
import
re
import
datetime
import
argparse
from
pathlib
import
Path
from
eval_utils
import
eval_utils
from
pcdet.config
import
cfg
,
cfg_from_list
,
cfg_from_yaml_file
,
log_config_to_file
from
pcdet.datasets
import
build_dataloader
from
pcdet.models
import
build_network
from
pcdet.utils
import
common_utils
from
pcdet.config
import
cfg
,
cfg_from_list
,
cfg_from_yaml_file
,
log_config_to_file
from
eval_utils
import
eval_utils
def
parse_config
():
...
...
@@ -195,4 +197,3 @@ def main():
if
__name__
==
'__main__'
:
main
()
tools/train.py
View file @
44c10a3a
import
argparse
import
datetime
import
glob
import
os
from
pathlib
import
Path
from
test
import
repeat_eval_ckpt
import
torch
import
torch.distributed
as
dist
import
torch.nn
as
nn
from
tensorboardX
import
SummaryWriter
from
pcdet.config
import
cfg
,
log_config_to_file
,
cfg_from_list
,
cfg_from_yaml_file
from
pcdet.
utils
import
c
ommon_ut
il
s
from
pcdet.
config
import
c
fg
,
cfg_from_list
,
cfg_from_yaml_file
,
log_config_to_f
il
e
from
pcdet.datasets
import
build_dataloader
from
pcdet.models
import
build_network
,
model_fn_decorator
from
pcdet.utils
import
common_utils
from
train_utils.optimization
import
build_optimizer
,
build_scheduler
from
train_utils.train_utils
import
train_model
import
torch.distributed
as
dist
from
test
import
repeat_eval_ckpt
from
pathlib
import
Path
import
argparse
import
datetime
import
glob
def
parse_config
():
...
...
tools/train_utils/optimization/__init__.py
View file @
44c10a3a
from
functools
import
partial
import
torch.nn
as
nn
import
torch.optim
as
optim
import
torch.optim.lr_scheduler
as
lr_sched
from
functools
import
partial
from
.fastai_optim
import
OptimWrapper
from
.learning_schedules_fastai
import
OneCycle
,
CosineWarmupLR
from
.learning_schedules_fastai
import
CosineWarmupLR
,
OneCycle
def
build_optimizer
(
model
,
optim_cfg
):
...
...
tools/train_utils/optimization/fastai_optim.py
View file @
44c10a3a
...
...
@@ -4,8 +4,8 @@ from collections import Iterable
import
torch
from
torch
import
nn
from
torch.nn.utils
import
parameters_to_vector
from
torch._utils
import
_unflatten_dense_tensors
from
torch.nn.utils
import
parameters_to_vector
bn_types
=
(
nn
.
BatchNorm1d
,
nn
.
BatchNorm2d
,
nn
.
BatchNorm3d
,
nn
.
SyncBatchNorm
)
...
...
tools/train_utils/optimization/learning_schedules_fastai.py
View file @
44c10a3a
# This file is modified from https://github.com/traveller59/second.pytorch
import
numpy
as
np
import
math
from
functools
import
partial
import
numpy
as
np
import
torch.optim.lr_scheduler
as
lr_sched
from
.fastai_optim
import
OptimWrapper
...
...
tools/train_utils/train_utils.py
View file @
44c10a3a
import
torch
import
os
import
glob
import
os
import
torch
import
tqdm
from
torch.nn.utils
import
clip_grad_norm_
...
...
tools/visual_utils/visualize_utils.py
View file @
44c10a3a
import
numpy
as
np
import
mayavi.mlab
as
mlab
import
numpy
as
np
import
torch
box_colormap
=
[
...
...
Prev
1
2
3
4
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment