You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The example in the Quick Tour section of the README file goes as follows:
fromaiseeimportTrainer, VisionClassifier# Initialize a VisionClassifier modelmodel=VisionClassifier(
model_name="vgg11_bn",
num_classes=4,
)
# Create a Trainer trainer=Trainer(
base_model=model,
data=f"animals/",
output_dir="trained_weights.pt",
)
# Traintrainer.train()
However, this example assumes that the data folder animals/ contains a subfolder named train with the training data, and another folder val with the validation data. This requirement is correctly documented in the Trainer API, but I believe it should also be stated in the README for clarity.
The text was updated successfully, but these errors were encountered:
The example in the Quick Tour section of the README file goes as follows:
However, this example assumes that the data folder
animals/
contains a subfolder namedtrain
with the training data, and another folderval
with the validation data. This requirement is correctly documented in the Trainer API, but I believe it should also be stated in the README for clarity.The text was updated successfully, but these errors were encountered: