QI JUN

Training loop part: Go+Python ```go for epoch := 0; epoch < niter; epoch++ { for i, data := range dataloader { netD.ZeroGrad() realCpu := data[0].To(device) batchSize := realCpu.Size(0) label :=...

Go+语法建议： 1. struct的定义与Python相比略显啰嗦，Go中 struct的定义 和 方法的定义是分离的，Python中是在一起的，并且Python中，成员变量不用提前声明。 2. Python中允许有些变量定义了，但没有被使用，这在Go中会强制报错，使用Python的算法同学可能一时不习惯。 3. `for i in range(10)` 相比于 `for i := 0; i < 10; i++` 还是要简洁一些，希望Go+同样支持 4. Python的tuple数据类型在Go+中如何支持？Go是不支持tuple的 5. `nn.ConvTranspose2d(ngf * 8, ngf...

@wangkuiyi >上面例子里什么地方需要 tuple 了？ 在这里 `optimizerG = optim.Adam(netG.parameters(), lr=opt.lr, betas=(opt.beta1, 0.999))` ，betas是一个tuple类型

@ganliqiang Could you please share your commands?

Hi @Lokiiiiii , Sorry for the later response. Thanks for submitting the MR and really appreciate your contributions to TensorRT-LLM. Could you please rebase the MR to the latest main...

> @QiJune Could you please review this again ? It LGTM now. We plan to integrate your contributions as part of our refinement work and when the work gets landed...

Please install the requirements.txt in the llama example first: ```python pip install -r examples/llama/requirements.txt ```