Skip to content
Extraits de code Groupes Projets
Valider e557f2a0 rédigé par Victor Joos de ter Beerst's avatar Victor Joos de ter Beerst
Parcourir les fichiers

Other task

parent 2f8c1751
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
_target_: flash.image.ImageClassificationData.from_csv
input_field: "Id"
target_fields: ["Action", "Romance", "Crime", "Thriller", "Adventure"]
train_file: ${preprocess.path}/movie_posters/train/metadata.csv
train_resolver:
_target_: hydra_flash.utils.resolver
_partial_: true
val_file: ${preprocess.path}/movie_posters/val/metadata.csv
val_resolver:
_target_: hydra_flash.utils.resolver
_partial_: true
transform_kwargs:
image_size: [128, 128]
batch_size: 1
_target_: flash.image.ImageClassificationData.from_files
predict_files:
- "${preprocess.path}/movie_posters/predict/tt0085318.jpg"
- "${preprocess.path}/movie_posters/predict/tt0089461.jpg"
- "${preprocess.path}/movie_posters/predict/tt0097179.jpg"
batch_size: 3
# @package _global_
defaults:
- _self_
- /server: local
- /preprocess: movie_poster
- /datamodule: movie_posters
- /model: resnet18
- /datamodule@predict_datamodule: movie_posters_test
trainer:
_target_: flash.Trainer
max_epochs: 3
gpus: 1
defaults:
- download
url: "https://pl-flash-data.s3.amazonaws.com/movie_posters.zip"
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Terminez d'abord l'édition de ce message.
Veuillez vous inscrire ou vous pour commenter