@kl_divergence   I just noticed your problem and will keep track on it. Hope that you can solve your problem soon.
@all I’m sorry to interrupt your conversation. I am wondering why we use setattr method?
setattr(self,"fc0",augmented1(2048))
setattr(self,"fc1",augmented2(2048))

but not like the normal way below:
self.fc0 = augmented1(2048)
self.fc1 = augmented2(2048)

Is there any difference between two above implementation for network definition?
Thanks.