(src_vars, dst_vars)
| 51 | |
| 52 | # Make a Tensorflow op to copy weights from one scope to another |
| 53 | def get_copy_params_op(src_vars, dst_vars): |
| 54 | src_vars = list(sorted(src_vars, key=lambda v: v.name)) |
| 55 | dst_vars = list(sorted(dst_vars, key=lambda v: v.name)) |
| 56 | |
| 57 | ops = [] |
| 58 | for s, d in zip(src_vars, dst_vars): |
| 59 | op = d.assign(s) |
| 60 | ops.append(op) |
| 61 | |
| 62 | return ops |
| 63 | |
| 64 | |
| 65 | def make_train_op(local_net, global_net): |