-
Notifications
You must be signed in to change notification settings - Fork 2.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable ParallelExecutor in SSD-MobileNet and Refine code. #832
Conversation
batch_size, | ||
num_passes, | ||
model_save_dir, | ||
pretrained_model=None): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
parallel_do
need have an argument to define using nccl
or not. The default value does not use nccl
.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done.
fluid/object_detection/train.py
Outdated
return os.path.exists(os.path.join(pretrained_model, var.name)) | ||
fluid.io.load_vars(exe, pretrained_model, predicate=if_exist) | ||
|
||
train_exe = fluid.ParallelExecutor(use_cuda=True, loss_name=loss.name) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
line 218 and line 227 is not consistent, currently ParallelExecutor
only allows to use GPU.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Here change use_cuda=arg.use_gpu
. Thanks!
for _, data in enumerate(test_reader()): | ||
test_map = exe.run(test_program, | ||
feed=feeder.feed(data), | ||
fetch_list=[accum_map]) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
exe.run
is not the format of parallel_exe
.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The testing use Executor, not ParallelExecutor.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
No description provided.