Nicki Minaj Model (#28218)
* New model: 2ff7490f-8a2f-4c0d-87a8-b0fa3a4a6a71 * Updated model_replay_ref_commit old-commit-hash: 328b5d9d47ef24140b316c375665e1cfd00cef31
This commit is contained in:
@@ -1,10 +1,5 @@
|
||||
#pragma once
|
||||
|
||||
// gate this here
|
||||
#define TEMPORAL
|
||||
#define DESIRE
|
||||
#define TRAFFIC_CONVENTION
|
||||
|
||||
#include <array>
|
||||
#include <memory>
|
||||
|
||||
@@ -17,6 +12,12 @@
|
||||
#include "selfdrive/modeld/models/nav.h"
|
||||
#include "selfdrive/modeld/runners/run.h"
|
||||
|
||||
// gate this here
|
||||
#define TEMPORAL
|
||||
#define DESIRE
|
||||
#define TRAFFIC_CONVENTION
|
||||
#define NAV
|
||||
|
||||
constexpr int FEATURE_LEN = 128;
|
||||
constexpr int HISTORY_BUFFER_LEN = 99;
|
||||
constexpr int DESIRE_LEN = 8;
|
||||
|
||||
Binary file not shown.
@@ -41,13 +41,13 @@ void ThneedModel::addExtra(float *extra_input_buf, int buf_size) {
|
||||
}
|
||||
|
||||
void* ThneedModel::getInputBuf() {
|
||||
if (use_extra && thneed->input_clmem.size() > 4) return &(thneed->input_clmem[4]);
|
||||
else if (!use_extra && thneed->input_clmem.size() > 3) return &(thneed->input_clmem[3]);
|
||||
if (use_extra && thneed->input_clmem.size() > 5) return &(thneed->input_clmem[5]);
|
||||
else if (!use_extra && thneed->input_clmem.size() > 4) return &(thneed->input_clmem[4]);
|
||||
else return nullptr;
|
||||
}
|
||||
|
||||
void* ThneedModel::getExtraBuf() {
|
||||
if (thneed->input_clmem.size() > 3) return &(thneed->input_clmem[3]);
|
||||
if (thneed->input_clmem.size() > 4) return &(thneed->input_clmem[4]);
|
||||
else return nullptr;
|
||||
}
|
||||
|
||||
@@ -55,10 +55,10 @@ void ThneedModel::execute() {
|
||||
if (!recorded) {
|
||||
thneed->record = true;
|
||||
if (use_extra) {
|
||||
float *inputs[5] = {recurrent, trafficConvention, desire, extra, input};
|
||||
float *inputs[6] = {recurrent, navFeatures, trafficConvention, desire, extra, input};
|
||||
thneed->copy_inputs(inputs);
|
||||
} else {
|
||||
float *inputs[4] = {recurrent, trafficConvention, desire, input};
|
||||
float *inputs[5] = {recurrent, navFeatures, trafficConvention, desire, input};
|
||||
thneed->copy_inputs(inputs);
|
||||
}
|
||||
thneed->clexec();
|
||||
@@ -68,12 +68,11 @@ void ThneedModel::execute() {
|
||||
recorded = true;
|
||||
} else {
|
||||
if (use_extra) {
|
||||
float *inputs[5] = {recurrent, trafficConvention, desire, extra, input};
|
||||
float *inputs[6] = {recurrent, navFeatures, trafficConvention, desire, extra, input};
|
||||
thneed->execute(inputs, output);
|
||||
} else {
|
||||
float *inputs[4] = {recurrent, trafficConvention, desire, input};
|
||||
float *inputs[5] = {recurrent, navFeatures, trafficConvention, desire, input};
|
||||
thneed->execute(inputs, output);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -1 +1 @@
|
||||
9d3cd2e7d5fceaaf0e8a4bd798a24fcf470da7c2
|
||||
e3cf1856830902bede2d1c9ca3d0d60e5504ae20
|
||||
|
||||
Reference in New Issue
Block a user