Блог

Я работал над генетическим алгоритмом, в котором я пытаюсь взять лучшую модель моей популяции (на основе оценки) и превратить всю популяцию в эту модель. Я сделал это успешно, но когда я пытаюсь мутировать каждую модель отдельно, все модели мутируют с одинаковыми параметрами. Я знаю, что это потому, что я использую объект и просто клонирую его в список, но я не знаю, что изменить, чтобы он не работал таким образом. Ниже я привел воспроизводимый пример для запуска в Python 3.9. Я знаю, что код не особенно мал, но это все, что я могу сделать. Заранее спасибо, любая помощь будет признательна.

 import torch
import torch.nn as nn

torch.manual_seed(0) #Reproducibility

population_size = 3 #Defining the size of my population
population = [nn.Linear(1,1) for _ in range(population_size)] #Initializing the population
input = torch.rand(1)# Creating dummy input
def mutate(m): #Function to mutate a model
    if type(m) == nn.Linear:
        m.weight = nn.Parameter(m.weight torch.randn(m.weight.shape))
        m.bias = nn.Parameter(m.bias torch.randn(m.bias.shape))

for i in population:
    print (i(input))

population = [x.apply(mutate) for x in population]

print ('n')

for i in population:
    print (i(input))
#The above works as expected

#I want to fill my entire population with that model.
#I've been filling the population with the best model by doing the following:

best_model = population[0] #Say the first model in the list was the best performing one

population = [best_model for _ in range(population_size)] #This is the line I think needs to change, I just don't know what to change it to. 

#This does fill the population with my best model, but when I try to mutate it, every model is mutated to the same parameters

population = [x.apply(mutate) for x in population] #I know this is because I am using best_model while replacing the population, but I don't know how else to copy the model

for i in population:
    print (i(input)) #Just to show that the population all gives the same result
 

Вы можете сделать глубокую копию модели. Убедитесь в import copy этом , а затем измените

 population = [best_model for _ in range(population_size)]
 

Для

 population = [copy.deepcopy(best_model) for _ in range(population_size)]