Closed gaozhihan closed 1 year ago
The unittest has passed, which means the new nn.Module
load the new state_dict
correctly.
https://github.com/amazon-science/earth-forecasting-transformer/blob/a01ea56de4baddc5b381757e5d789f7f1efdcffe/tests/unittests/test_pretrained_checkpoints.py#L171-L173
z_proj
.