221 Star 947 Fork 694

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
DgmrNet.yaml 525 Bytes
一键复制 编辑 原始数据 按行查看 历史
Liu Ruoyan 提交于 2024-12-12 11:25 +08:00 . 修改ipynb文件,fix vitknobug,fix jenkins
model:
name: DgmrNet
forecast_steps: 18
in_channels: 1
out_channels: 256
conv_type: "standard"
num_samples: 6
grid_lambda: 20.0
latent_channels: 768
context_channels: 384
generation_steps: 1
num_spatial_frames: 8
data:
name: "radar"
root_dir: "./dataset/"
batch_size: 1
optimizer:
name: "adam"
gen_lr: 0.00005
disc_lr: 0.0002
beta1: 0.0001
beta2: 0.999
loss_scale: 512
summary:
csi_thresholds: 0.3
save_checkpoint_epochs: 1
keep_checkpoint_max: 5
train:
name: "fp"
epochs: 20
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
r0.7

搜索帮助