Downloads · 30 days
6
32% of all-time downloads
Giux22/semana3-patrones-concentricos-1
semana3-patrones-concentricos-1 is a machine learning model from Giux22. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
imagesize = 128 the generated image resolution trainbatchsize = 16 evalbatchsize = 16 how many images to sample during evaluation numepochs = 15000 gradientaccumulationsteps = 1 learningrate = 1e-5 lrwarmupsteps = 500…
Downloads · 30 days
6
32% of all-time downloads
All-time downloads
19
Public
Repo size
273 MB
Likes
0
Public
Click a slice to open those files.
.png244 MB · 100%
From the Hugging Face model README
image_size = 128 # the generated image resolution train_batch_size = 16 eval_batch_size = 16 # how many images to sample during evaluation num_epochs = 15000 gradient_accumulation_steps = 1 learning_rate = 1e-5 lr_warmup_steps = 500 save_image_epochs = 50 save_model_epochs = 50
block_out_channels=(128, 128, 256, 256, 512, 512), # the number of output channels for each UNet block down_block_types=( "DownBlock2D", # a regular ResNet downsampling block "DownBlock2D", "DownBlock2D", "DownBlock2D", "AttnDownBlock2D", # a ResNet downsampling block with spatial self-attention "DownBlock2D", ),
preprocess = transforms.Compose( [ transforms.Resize((config.image_size, config.image_size)), transforms.RandomRotation(45), transforms.RandomHorizontalFlip(), transforms.RandomVerticalFlip(), transforms.ToTensor(), transforms.Normalize([0.5], [0.5]), ] )