Closed bennyguo closed 2 years ago
The dummy variable in Grid is defined as parameter, which is not used in training but may cause "unused parameter error" in some frameworks (for example when using DDP in pytorch-lightning).
LGTM thanks Yuanchen!
The dummy variable in Grid is defined as parameter, which is not used in training but may cause "unused parameter error" in some frameworks (for example when using DDP in pytorch-lightning).