Skip to content
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.

Sign up for GitHub

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

Jump to bottom

一般需要训练多久呢 #25

Open
sxmayd opened this issue Aug 7, 2022 · 5 comments
Open

一般需要训练多久呢 #25

sxmayd opened this issue Aug 7, 2022 · 5 comments

Comments

@sxmayd
Copy link

sxmayd commented Aug 7, 2022

请问大家一般训练需要多久呢，能否使用GPU进行加速？
@huangkk99
Copy link

原来我们使用tensorflow是比较慢的，可能需要1天时间。如果用pytorch改写之后，运行MARL需要约30-40分钟左右。可以使用GPU加速，但是因为我们使用的是全连接层，效果不会很明显。

@romjiangzexu
Copy link

可以分享一下pytorch版本吗

@huangkk99
Copy link

huangkk99 commented May 8, 2023 •
edited
Loading

可以参考一下https://github.com/huangkk99/MARLV2X。
但在其中训练MARL时有时不太稳定。

@romjiangzexu
Copy link

非常感谢

@HongShi007
Copy link

请问训练模型用到的数据集在哪个文件夹里呀

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@huangkk99 @sxmayd @HongShi007 @romjiangzexu