mirror of
https://github.com/babalae/better-genshin-impact.git
synced 2026-04-08 12:30:42 +08:00
* 增加RodNet必须支持训练的单元测试;RodNet的torch链路改造成全张量计算,由此把之前忽略的参数dz、h_coeff、offset变得可学习 * 了解到损失函数CrossEntropyLoss内置了softmax,因此从forward方法中移出;offset是手动指定的偏置值,不是学习得到的,因此也移出到单独的PostProcess方法中 * 根据得到的源码整理RodNet;新增在数据集上达到一定准确率的单元测试
16 lines
353 B
C#
16 lines
353 B
C#
using static TorchSharp.torch;
|
|
|
|
namespace BetterGenshinImpact.GameTask.AutoFishing;
|
|
|
|
public record RodInput
|
|
{
|
|
public double rod_x1;
|
|
public double rod_x2;
|
|
public double rod_y1;
|
|
public double rod_y2;
|
|
public double fish_x1;
|
|
public double fish_x2;
|
|
public double fish_y1;
|
|
public double fish_y2;
|
|
public int fish_label;
|
|
} |