Closed ObcTech closed 1 month ago
Hi @ObcTech
You can test the model using the below command:
python3 main.py test --config config/efficientnetv2_s/cifar10.yaml --trainer.gpus 2,3,
You can easily implement weight loading like below code snippet:
# Line 44 in main.py (https://github.com/hankyul2/EfficientNetV2-pytorch/blob/main/main.py)
self.model = get_efficientnet_v2(model_name, pretrained, num_classes, dropout=dropout)
state_dict = torch.load(checkpoint_path)
self.model.load_state_dict(state_dict)
Thank you.
Hankyul.
same as title ↑↑↑↑↑↑↑