Downloads · 30 days
0
Giux22/semana6-patrones_concentricos_con_fondo
semana6-patrones_concentricos_con_fondo 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.
numepochs = 15000 gradientaccumulationsteps = 1 learningrate = 7e-5 lrwarmupsteps = 500
Downloads · 30 days
0
Access
Public
Updated Apr 16, 2024
Repo size
34.9 MB
Likes
0
Public
Click a slice to open those files.
.png34.9 MB · 100%
From the Hugging Face model README
num_epochs = 15000 gradient_accumulation_steps = 1 learning_rate = 7e-5 lr_warmup_steps = 500
[ transforms.Resize((config.image_size, config.image_size)), transforms.RandomRotation(30), transforms.RandomHorizontalFlip(), transforms.RandomVerticalFlip(), transforms.ToTensor(), transforms.Normalize([0.5], [0.5]), ]
sample_size=config.image_size, # the target image resolution in_channels=3, # the number of input channels, 3 for RGB images out_channels=3, # the number of output channels layers_per_block=2, # how many ResNet layers to use per UNet block 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", ),