DeepSense Forum

Clear all

[Solved] Question About Radar-Aided Beam Prediction

New Member
Joined: 3 months ago
Posts: 1
Topic starter  

Hi DeepSense Team...

Thank you for your greatwork.

I am interested to run radar-aided-beam-prediction-repo. I see that there are functions called train_loop(), test_loop(), and eval_loop() inside How to call these functions? I am still don't have idea what I should put in this paramaters: criterion and optimizer.

I see in the paper the authors using lookup table mapping as the baseline algorithm, however I can't find the baseline code in the repository. Could you show me where the code is?

Thank you

This topic was modified 1 month ago by avn

Member Admin
Joined: 2 years ago
Posts: 1

Hi @avn,

You can use the classical definitions for the loss function (criterion) and optimizer. In this paper, we used Adam optimizer with cross-entropy loss, which can be defined as follows

criterion = torch.nn.CrossEntropyLoss() # Mean for training
optimizer = torch.optim.Adam(net.parameters(), lr=1e-3, weight_decay=1e-4)

Note that the variable "net" is the neural network object. For the baseline, we have added the script named to the repository to generate the lookup table solution results.



avn liked

Leave a reply

Author Name

Author Email

Title *

Preview 0 Revisions Saved