|
|
|
|
|
by albertzeyer
1649 days ago
|
|
Well, almost exactly the same would work in TensorFlow as well. But the author choose to make the TF logic more explicitly visible, and use abstractions like tf.Module (like torch.nn.Module), the dataset, etc. I'm not sure this really makes so much sense. I guess this is more for just playing around. |
|